Update my_model/state_manager.py
Browse files
my_model/state_manager.py
CHANGED
@@ -44,7 +44,7 @@ class StateManager:
|
|
44 |
"""
|
45 |
Sets up user interface widgets for selecting models, settings, and displaying model settings conditionally.
|
46 |
"""
|
47 |
-
|
48 |
self.col1.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method', disabled=self.is_widget_disabled)
|
49 |
detection_model = self.col1.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model', disabled=self.is_widget_disabled)
|
50 |
default_confidence = 0.2 if st.session_state.detection_model == "yolov5" else 0.4
|
|
|
44 |
"""
|
45 |
Sets up user interface widgets for selecting models, settings, and displaying model settings conditionally.
|
46 |
"""
|
47 |
+
|
48 |
self.col1.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method', disabled=self.is_widget_disabled)
|
49 |
detection_model = self.col1.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model', disabled=self.is_widget_disabled)
|
50 |
default_confidence = 0.2 if st.session_state.detection_model == "yolov5" else 0.4
|