Spaces:
Sleeping
Sleeping
Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
|
@@ -155,7 +155,7 @@ class InferenceRunner(StateManager):
|
|
| 155 |
nested_col11, nested_col12 = st.columns([0.5, 0.5])
|
| 156 |
if nested_col11.button(st.session_state.button_label, on_click=self.disable_widgets, disabled=self.is_widget_disabled):
|
| 157 |
if st.session_state.button_label == "Load Model":
|
| 158 |
-
if self.is_model_loaded
|
| 159 |
free_gpu_resources()
|
| 160 |
fine_tuned_model_already_loaded = True
|
| 161 |
else:
|
|
|
|
| 155 |
nested_col11, nested_col12 = st.columns([0.5, 0.5])
|
| 156 |
if nested_col11.button(st.session_state.button_label, on_click=self.disable_widgets, disabled=self.is_widget_disabled):
|
| 157 |
if st.session_state.button_label == "Load Model":
|
| 158 |
+
if self.is_model_loaded:
|
| 159 |
free_gpu_resources()
|
| 160 |
fine_tuned_model_already_loaded = True
|
| 161 |
else:
|