Update app.py

app.py

@@ -57,14 +57,14 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
     
     if message["files"]:
         image = Image.open(message["files"][-1])
-        conversation.append({"role": "user", "content": f"<|image_1|>\n{message['text']}"})
     else:
         if len(history) == 0:
             raise gr.Error("Please upload an image first.")
             image = None
         elif len(history):
-            image = history        
-        conversation.append({"role": "user", "content": message['text']})
+            image = Image.open(history[0][0][0])
+    
+    conversation.append({"role": "user", "content": f"<|image_1|>\n{message['text']}"})
     print(f"Conversation is -\n{conversation}")
     inputs = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
     inputs_ids = processor(inputs, image, return_tensors="pt").to(0)