Update app.py

app.py

@@ -71,17 +71,25 @@ def infer(audio_file):
 
     result = client.predict(
     				llama_q,	# str in 'Message' Textbox component
-    				api_name="/chat_1"
+    				api_name="/chat"
     )
+
+    # will return a list of outputs, one for each iteration
+    while not result.done():
+        continue
+    
+    # will return a list of outputs, one for each iteration. 
+    # so get the last one
+    final_res = result.outputs()[-1]
     
-    print(result)
+    print(final_res)
 
-    images = pipe(prompt=result).images[0]
+    images = pipe(prompt=final_res).images[0]
 
     print("Finished")
     
     #return cap_result, result, images
-    return images, result, gr.update(visible=True)
+    return images, final_res, gr.update(visible=True)
 
 css = """
 #col-container {max-width: 510px; margin-left: auto; margin-right: auto;}