m7mdal7aj commited on
Commit
e36aa58
1 Parent(s): 2111435

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