Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -292,14 +292,15 @@ 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
-                    
-            st.write(st.session_state['method'])
-            st.write(st.session_state['previous_state']['method'])
             
             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 '
                     f'high-end GPUs, implementation will be explored in the future.')
+
+            st.write(st.session_state['method'])
+            st.write(st.session_state['previous_state']['method'])
+            
         if self.is_model_loaded:
             free_gpu_resources()
             st.session_state['loading_in_progress'] = False