Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -17,7 +17,6 @@ from my_model.state_manager import StateManager
 from my_model.config import inference_config as config
 
 
-
 class InferenceRunner(StateManager):
     """
     Manages the user interface and interactions for running inference using the Streamlit-based Knowledge-Based Visual
@@ -244,16 +243,15 @@ class InferenceRunner(StateManager):
         reload_kbvqa = False
         reload_detection_model = False
         force_reload_full_model = False
-        
 
         if self.is_model_loaded and self.settings_changed:
             self.col1.warning("Model settings have changed, please reload the model, this will take a second .. ")
-           # self.update_prev_state()
+        # self.update_prev_state()
         st.session_state.button_label = (
             "Reload Model" if (self.is_model_loaded and
                                st.session_state.kbvqa.detection_model != st.session_state['detection_model']) or
-                               (st.session_state['previous_state']['method'] is not None and 
-                                st.session_state['method'] != st.session_state['previous_state']['method'])
+                              (st.session_state['previous_state']['method'] is not None and
+                               st.session_state['method'] != st.session_state['previous_state']['method'])
             else "Load Model"
         )
 
@@ -269,10 +267,11 @@ class InferenceRunner(StateManager):
                                 fine_tuned_model_already_loaded = True
                             else:
                                 load_fine_tuned_model = True
-                        elif st.session_state.button_label == "Reload Model" 
-                            and st.session_state['method'] != st.session_state['previous_state']['method']: # check if the model size have changed
+                        elif st.session_state.button_label == "Reload Model" and st.session_state['method'] != \
+                                st.session_state['previous_state']['method']:  # check if the model size have changed
                             force_reload_full_model = True
-                        elif (self.is_model_loaded and st.session_state.kbvqa.detection_model != st.session_state['detection_model']):
+                        elif (self.is_model_loaded and st.session_state.kbvqa.detection_model != 
+                              st.session_state['detection_model']):
                             reload_detection_model = True
                     if nested_col12.button("Force Reload", on_click=self.disable_widgets,
                                            disabled=self.is_widget_disabled):
@@ -298,7 +297,7 @@ class InferenceRunner(StateManager):
                     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
-            
+
             elif st.session_state.method == "Vision-Language Embeddings Alignment":
                 self.col1.warning(
                     f'Model using {st.session_state.method} is desgined but requires large scale data and multiple '
@@ -308,7 +307,7 @@ class InferenceRunner(StateManager):
             st.write(st.session_state['previous_state']['method'])
             if st.session_state['kbvqa'] is not None:
                 st.write(st.session_state['kbvqa'].kbvqa_model_name)
-            
+
         if self.is_model_loaded:
             free_gpu_resources()
             st.session_state['loading_in_progress'] = False