KB-VQA-E

Sleeping

m7mdal7aj commited on May 15

Commit

bb9ebd1

•

1 Parent(s): 78c10a6

Update my_model/tabs/run_inference.py

Files changed (1) hide show

my_model/tabs/run_inference.py CHANGED Viewed

@@ -305,5 +305,6 @@ class InferenceRunner(StateManager):
         if self.is_model_loaded:
             free_gpu_resources()
             st.session_state['loading_in_progress'] = False
             self.image_qa_app()  # this is the main Q/A Application

         if self.is_model_loaded:
             free_gpu_resources()
             st.session_state['loading_in_progress'] = False
+            self.update_prev_state()
             self.image_qa_app()  # this is the main Q/A Application