Update app.py

app.py

@@ -59,10 +59,11 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
         if len(history) == 0:
             raise gr.Error("Please upload an image first.")
             image = None
-        elif len(history):
+        else:
             image = Image.open(history[0][0][0])
-            conversation.extend([{"role": "user", "content":""}])
             for prompt, answer in history:
+                if answer is None:
+                    conversation.extend([{"role": "user", "content":""}])
                 conversation[0]['content'] = f"<|image_1|>\n{prompt}"
                 conversation.extend([{"role": "assistant", "content": answer}])
             conversation.append({"role": "user", "content": message['text']})