Update my_model/utilities/state_manager.py
Browse files
my_model/utilities/state_manager.py
CHANGED
@@ -58,7 +58,7 @@ class StateManager:
|
|
58 |
#self.update_model_settings(detection_model, confidence_level)
|
59 |
# Update the previous state with current session state values
|
60 |
st.session_state['previous_state'] = {'method': st.session_state.method, 'detection_model': st.session_state.detection_model, 'confidence_level': st.session_state.confidence_level}
|
61 |
-
st.session_state['
|
62 |
free_gpu_resources()
|
63 |
except Exception as e:
|
64 |
st.error(f"Error loading model: {e}")
|
|
|
58 |
#self.update_model_settings(detection_model, confidence_level)
|
59 |
# Update the previous state with current session state values
|
60 |
st.session_state['previous_state'] = {'method': st.session_state.method, 'detection_model': st.session_state.detection_model, 'confidence_level': st.session_state.confidence_level}
|
61 |
+
st.session_state['button_lablel'] = "Reload Model"
|
62 |
free_gpu_resources()
|
63 |
except Exception as e:
|
64 |
st.error(f"Error loading model: {e}")
|