Update app.py

app.py

@@ -52,16 +52,17 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
     print(message)
     conversation = []
     for prompt, answer in history:
-        conversation.extend([{"role": "user", "content": f"<|image_1|>\n{prompt}"}, {"role": "assistant", "content": answer}])
-    conversation.append({"role": "user", "content": message['text']})
+        conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
     
     if message["files"]:
-        image = Image.open(message["files"][-1]).convert('RGB')
+        image = Image.open(message["files"][-1])
+        conversation.append({"role": "user"}, "content": f"<|image_1|>\n{message['text']}")
     else:
         if len(history) == 0:
             gr.Error("Please upload an image first.")
         image = None
-        
+        conversation.append({"role": "user", "content": message['text']})
+       
     prompt = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
     inputs = processor(prompt, images=image, return_tensors="pt").to(0)