Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -58,7 +58,6 @@ class InferenceRunner(StateManager):
         with self.col2:
             for image_key, image_data in self.get_images_data().items():
                 with st.container():
-                    self.display_session_state()
                     nested_col21, nested_col22 = st.columns([0.65, 0.35])
                     image_for_display = self.resize_image(image_data['image'], 600)
                     nested_col21.image(image_for_display, caption=f'Uploaded Image: {image_key[-11:]}')
@@ -95,9 +94,33 @@ class InferenceRunner(StateManager):
     def display_message(self, message, warning=False, write=False, text=False):
         pass
 
+
+
+    
+    
+    @st.experimental_memo(show_spinner=False)
+    def trigger_rerun(self):
+        pass  # No need for any specific code here
+    
+
+
+
         
     def run_inference(self):
+
+
         
+        if st.button("Re-run App"):
+            self.trigger_rerun(None)  # Force a re-run
+        else:
+            # Trigger re-run every 5 seconds (example)
+            if time.time() - st.session_state.get("last_rerun", 0) > 5:
+                self.trigger_rerun(None)
+                st.session_state["last_rerun"] = time.time()
+
+
+                
+            
         self.set_up_widgets()
         load_fine_tuned_model = False
         fine_tuned_model_already_loaded = False
@@ -137,7 +160,7 @@ class InferenceRunner(StateManager):
                     t1=time.time()
                     free_gpu_resources()
                     self.load_model()
-                    st.session_state['time_taken_to_load_model'] = time.time()-t1
+                    st.session_state['time_taken_to_load_model'] = int(time.time()-t1)
                     st.session_state['loading_in_progress'] = False
                     
                 elif fine_tuned_model_already_loaded:
@@ -154,7 +177,7 @@ class InferenceRunner(StateManager):
                     free_gpu_resources()
                     t1=time.time()
                     self.force_reload_model()
-                    st.session_state['time_taken_to_load_model'] = time.time()-t1
+                    st.session_state['time_taken_to_load_model'] = int(time.time()-t1)
                     st.session_state['loading_in_progress'] = False
                     st.session_state['model_loaded'] = True