Update app.py

app.py

@@ -53,28 +53,19 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
     print(f'history is - {history}')
     conversation = [] 
     if message["files"]:
-        image_path = message["files"][-1]
-        if isinstance(image_path, dict):
-            mage_path = image_path["path"]
-        image = Image.open(image_path)
+        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
         else:
-            if isinstance(history[0][0], tuple):
-                image_path = history[0][0][0]
-            else:
-                image_path = history[0][0]
-            image = Image.open(image_path)
+            image = Image.open(history[0][0][0])
             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": "user", "content":"<|image_1|>"},{"role": "assistant", "content": ""}])
                 conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
-        conversation.append({"role": "user", "content": message['text']})
-    
+            conversation.append({"role": "user", "content": 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)