Update app.py

app.py

@@ -50,7 +50,7 @@ def handle_image_submission(_chatbot, task_history, file) -> tuple:
     history_item = ((file_path,), None)
     _chatbot.append(history_item)
     task_history.append(history_item)
-    return predict(_chatbot, task_history, tokenizer, model)
+    return predict(_chatbot, task_history)
 
     
 def _load_model_tokenizer(args) -> tuple:
@@ -185,7 +185,7 @@ def regenerate(_chatbot, task_history) -> list:
         _chatbot[-1] = (_chatbot[-1][0], None)
     else:
         _chatbot.append((chatbot_item[0], None))
-    return predict(_chatbot, task_history, tokenizer, model)
+    return predict(_chatbot, task_history)
 
 def add_text(history, task_history, text) -> tuple:
     task_text = text