Update app.py

app.py

@@ -22,7 +22,7 @@ model.to("cuda:0")
 
 @spaces.GPU
 def bot_streaming(message, history):
-    chat_history = []
+    messages = []
     if message["files"]:
       image = message["files"][-1]["path"]
     else:
@@ -32,21 +32,21 @@ def bot_streaming(message, history):
           image_turn = i
             
     if len(history) > 0 and image is not None:
-        chat_history.append({"role": "user", "content": f'<image>\n{history[1][0]}'})
-        chat_history.append({"role": "assistant", "content": history[1][1] })
+        messages.append({"role": "user", "content": f'<image>\n{history[1][0]}'})
+        messages.append({"role": "assistant", "content": history[1][1] })
         for human, assistant in history[2:]:
-            chat_history.append({"role": "user", "content": human })
-            chat_history.append({"role": "assistant", "content": assistant })
-        chat_history.append({"role": "user", "content": message['text']})
+            messages.append({"role": "user", "content": human })
+            messages.append({"role": "assistant", "content": assistant })
+        messages.append({"role": "user", "content": message['text']})
     elif len(history) > 0 and image is None:
         for human, assistant in history:
-            chat_history.append({"role": "user", "content": human })
-            chat_history.append({"role": "assistant", "content": assistant })
-        chat_history.append({"role": "user", "content": message['text']})
+            messages.append({"role": "user", "content": human })
+            messages.append({"role": "assistant", "content": assistant })
+        messages.append({"role": "user", "content": message['text']})
     elif len(history) == 0 and image is not None:
-        chat_history.append({"role": "user", "content": f"<image>\n{message['text']}"})
+        messages.append({"role": "user", "content": f"<image>\n{message['text']}"})
     elif len(history) == 0 and image is None:
-        chat_history.append({"role": "user", "content": message['text'] })
+        messages.append({"role": "user", "content": message['text'] })
 
     # if image is None:
     #     gr.Error("You need to upload an image for LLaVA to work.")