Update app.py

app.py

@@ -35,7 +35,7 @@ def _get_args() -> ArgumentParser:
     args = parser.parse_args()
     return args
 
-def handle_image_submission(_chatbot, task_history, file) -> tuple:
+def handle_image_submission(_chatbot, task_history, file, tokenizer, model) -> tuple:
     print("handle_image_submission called")
     if file is None:
         print("No file uploaded")
@@ -251,32 +251,36 @@ Join us : 🌟TeamTonic🌟 is always making cool demos! Join our active builder
         submit_btn.click(
             fn=predict,
             inputs=[chatbot, task_history],
-            outputs=[chatbot]
+            outputs=[chatbot],
+            _state=[tokenizer, model]
         )
         
         submit_file_btn.click(
             fn=handle_image_submission,
             inputs=[chatbot, task_history, file_upload],
-            outputs=[chatbot, task_history]
+            outputs=[chatbot, task_history],
+            _state=[tokenizer, model]
         )
 
-
         regen_btn.click(
             fn=regenerate,
             inputs=[chatbot, task_history],
-            outputs=[chatbot]
+            outputs=[chatbot],
+            _state=[tokenizer, model]
         )
 
         empty_bin.click(
             fn=reset_state,
             inputs=[task_history],
-            outputs=[task_history]
+            outputs=[task_history],
+            _state=[tokenizer, model]
         )
 
         query.submit(
             fn=add_text,
             inputs=[chatbot, task_history, query],
-            outputs=[chatbot, task_history, query]
+            outputs=[chatbot, task_history, query],
+            _state=[tokenizer, model]
         )
 
         gr.Markdown("""