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:
|