Update my_model/utilities/state_manager.py
Browse files
my_model/utilities/state_manager.py
CHANGED
|
@@ -60,6 +60,7 @@ class StateManager:
|
|
| 60 |
|
| 61 |
st.session_state['button_label'] = "Reload Model"
|
| 62 |
st.text('button changed')
|
|
|
|
| 63 |
free_gpu_resources()
|
| 64 |
except Exception as e:
|
| 65 |
st.error(f"Error loading model: {e}")
|
|
|
|
| 60 |
|
| 61 |
st.session_state['button_label'] = "Reload Model"
|
| 62 |
st.text('button changed')
|
| 63 |
+
self.has_state_changed()
|
| 64 |
free_gpu_resources()
|
| 65 |
except Exception as e:
|
| 66 |
st.error(f"Error loading model: {e}")
|