Update app.py

app.py

@@ -57,8 +57,10 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
             image = None
         else:
             image = Image.open(history[0][0][0])
-            for prompt, answer in history:
-                conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
+                if answer is None:
+                    conversation.extend([{"role": "user", "content": prompt},{"role": "assistant", "content": ""}])
+                else:
+                    conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
             conversation.append({"role": "user", "content": message['text']})    
     print(f"Conversation is -\n{conversation}")