m7mdal7aj commited on
Commit
2111435
1 Parent(s): 7daf825

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +1 -1
my_model/tabs/run_inference.py CHANGED
@@ -195,7 +195,7 @@ class InferenceRunner(StateManager):
195
  self.col1.warning(f'Model using {st.session_state.method} is not deployed yet, will be ready later.')
196
  st.session_state['loading_in_progress'] = False
197
 
198
- if self.is_model_loaded():
199
  free_gpu_resources()
200
  st.session_state['loading_in_progress'] = False
201
  self.image_qa_app(self.get_model())
 
195
  self.col1.warning(f'Model using {st.session_state.method} is not deployed yet, will be ready later.')
196
  st.session_state['loading_in_progress'] = False
197
 
198
+ if self.is_model_loaded:
199
  free_gpu_resources()
200
  st.session_state['loading_in_progress'] = False
201
  self.image_qa_app(self.get_model())