Spaces:
Sleeping
Sleeping
Update my_model/tabs/run_inference.py
Browse files
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())
|