Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -103,21 +103,10 @@ class InferenceRunner(StateManager):
 
         
     def run_inference(self):
+        if st.session_state['loading_in_progress']:
+            st.rerun
+        self.set_up_widgets(rerun)
 
-
-        
-        if st.button("Re-run App"):
-            trigger_rerun()  # Force a re-run
-        else:
-            # Trigger re-run every 5 seconds (example)
-            if time.time() - st.session_state.get("last_rerun", 0) > 5:
-                trigger_rerun()
-                st.session_state["last_rerun"] = time.time()
-
-
-                
-            
-        self.set_up_widgets()
         load_fine_tuned_model = False
         fine_tuned_model_already_loaded = False
         reload_detection_model = False
@@ -189,11 +178,3 @@ class InferenceRunner(StateManager):
             self.image_qa_app(self.get_model())
             st.write('load success', 'loading in prog?', st.session_state['loading_in_progress'])
 
-
-
-
-
-    
-@st.experimental_memo(show_spinner=False)
-def trigger_rerun():
-    pass  # No need for any specific code here