Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -94,10 +94,6 @@ class InferenceRunner(StateManager):
         st.session_state.button_label = "Reload Model" if self.is_model_loaded() and self.settings_changed else "Load Model"
 
         with self.col1:
-            
-            if self.is_model_loaded():
-                free_gpu_resources()
-                self.image_qa_app(self.get_model())
                 
             if st.session_state.method == "Fine-Tuned Model":
                 
@@ -129,11 +125,17 @@ class InferenceRunner(StateManager):
                 elif reload_detection_model:
                     free_gpu_resources()
                     self.reload_detection_model()
+                    
                 elif force_reload_full_model:
                     free_gpu_resources()
                     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.')
+                self.col1.warning(f'Model using {st.session_state.method} is not deployed yet, will be ready later.')
+
+        
+        if self.is_model_loaded():
+            free_gpu_resources()
+            self.image_qa_app(self.get_model())
+
+