Update app.py

app.py

@@ -46,7 +46,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)
+    return predict(_chatbot, task_history, tokenizer, model)
 
     
 def _load_model_tokenizer(args) -> tuple:
@@ -126,7 +126,7 @@ def add_file(history, task_history, file):
     return history, task_history
 
 
-def predict(_chatbot, task_history) -> list:
+def predict(_chatbot, task_history, tokenizer, model) -> list:
     print("predict called")
     if not _chatbot:
         return _chatbot 
@@ -180,7 +180,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)
+    return predict(_chatbot, task_history, tokenizer, model)
 
 def add_text(history, task_history, text) -> tuple:
     task_text = text