Update app.py

app.py

@@ -127,16 +127,15 @@ def add_file(history, task_history, file):
 def predict(_chatbot, task_history) -> tuple:
     print("predict called")
     chat_query, chat_response = _chatbot[-1]
-
     if isinstance(chat_query, tuple):
-        query = [{'image': chat_query[0]}]
+        chat_query = chat_query[0]
+        query = [{'image': chat_query}]
     else:
         query = [{'text': _parse_text(chat_query)}]
-
+    
     inputs = tokenizer.from_list_format(query)
     tokenized_inputs = tokenizer(inputs, return_tensors='pt')
     tokenized_inputs = tokenized_inputs.to(model.device)
-
     pred = model.generate(**tokenized_inputs)
     response = tokenizer.decode(pred.cpu()[0], skip_special_tokens=False)
 
@@ -149,7 +148,7 @@ def predict(_chatbot, task_history) -> tuple:
             image_path = save_image(image, uploaded_file_dir)
             _chatbot[-1] = (chat_query, (image_path,))
         else:
-            _chatbot[-1] = (chat_query, "No image to display.")
+            _chatbot[-1] = (chat_query, response)  # Make sure this is a tuple of two elements
     else:
         _chatbot[-1] = (chat_query, response)