Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -118,18 +118,18 @@ class InferenceRunner(StateManager):
                         self.image_qa_app(self.get_model())
 
                 if load_fine_tuned_model:
-                    st.write("TEXTS")
                     free_gpu_resources()
                     self.load_model()
                     
                 elif fine_tuned_model_already_loaded:
+                    free_gpu_resources()
                     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:
+                    free_gpu_resources()
                     self.force_reload_model()
                 
             elif st.session_state.method == "In-Context Learning (n-shots)":