Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -115,19 +115,19 @@ class InferenceRunner(StateManager):
                         free_gpu_resources()
                         self.image_qa_app(self.get_model())
 
-            elif load_fine_tuned_model:
-                free_gpu_resources()
-                self.load_model()
-                
-            elif fine_tuned_model_already_loaded:
-                self.col1.text("Model already loaded and no settings were changed:)")      
-                
-            elif reload_detection_model:
-                free_gpu_resources()
-                
-                self.reload_detection_model()
-            elif force_reload_full_model:
-                self.force_reload_model()
+                elif load_fine_tuned_model:
+                    free_gpu_resources()
+                    self.load_model()
+                    
+                elif fine_tuned_model_already_loaded:
+                    self.col1.text("Model already loaded and no settings were changed:)")      
+                    
+                elif reload_detection_model:
+                    free_gpu_resources()
+                    
+                    self.reload_detection_model()
+                elif force_reload_full_model:
+                    self.force_reload_model()
                 
             elif st.session_state.method == "In-Context Learning (n-shots)":
                 self.col1.warning(f'Model using {st.session_state.method} is not deployed yet, will be ready later.')