Update app.py

app.py

@@ -118,7 +118,7 @@ def chat(message, history):
 We currently only support one image at the start of the context! Please start a new conversation."""
     
     if len(history) == 0 and len(message["files"]) != 0:
-        test_image = message["files"][0]["path"]
+        test_image = message["files"][0]
         pixel_values = load_image(test_image, max_num=6).to(torch.bfloat16).cuda()
     elif len(history) == 0 and len(message["files"]) == 0:
         pixel_values = None     
@@ -157,13 +157,13 @@ We currently only support one image at the start of the context! Please start a
         
     print(f'User: {question}\nAssistant: {response}')
 
-    return response
-    # buffer = ""
-    # for new_text in response:
-    #   buffer += new_text
-    #   generated_text_without_prompt = buffer[:]
-    #   time.sleep(0.005)
-    #   yield generated_text_without_prompt
+    # return response
+    buffer = ""
+    for new_text in response:
+      buffer += new_text
+      generated_text_without_prompt = buffer[:]
+      time.sleep(0.005)
+      yield generated_text_without_prompt
 
 CSS ="""
 # @media only screen and (max-width: 600px){