Update my_model/state_manager.py
Browse files
my_model/state_manager.py
CHANGED
@@ -64,8 +64,8 @@ class StateManager:
|
|
64 |
Sets up user interface widgets for selecting models, settings, and displaying model settings conditionally.
|
65 |
"""
|
66 |
|
67 |
-
self.col1.selectbox("Choose a model:", ["13b-Fine-Tuned Model", "7b-Fine-Tuned Model", "Vision-Language Embeddings Alignment"], index=
|
68 |
-
detection_model = self.col1.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=
|
69 |
default_confidence = 0.2 if st.session_state.detection_model == "yolov5" else 0.4
|
70 |
self.set_slider_value(text="Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.05, slider_key_name='confidence_level', col=self.col1)
|
71 |
|
|
|
64 |
Sets up user interface widgets for selecting models, settings, and displaying model settings conditionally.
|
65 |
"""
|
66 |
|
67 |
+
self.col1.selectbox("Choose a model:", ["13b-Fine-Tuned Model", "7b-Fine-Tuned Model", "Vision-Language Embeddings Alignment"], index=1, key='method', disabled=self.is_widget_disabled)
|
68 |
+
detection_model = self.col1.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=0, key='detection_model', disabled=self.is_widget_disabled)
|
69 |
default_confidence = 0.2 if st.session_state.detection_model == "yolov5" else 0.4
|
70 |
self.set_slider_value(text="Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.05, slider_key_name='confidence_level', col=self.col1)
|
71 |
|