Spaces:
Sleeping
Sleeping
Update my_model/utilities/state_manager.py
Browse files
my_model/utilities/state_manager.py
CHANGED
|
@@ -19,7 +19,7 @@ class StateManager:
|
|
| 19 |
if 'kbvqa' not in st.session_state:
|
| 20 |
st.session_state['kbvqa'] = None
|
| 21 |
|
| 22 |
-
self.default_confidence = 0.2 if detection_model == "yolov5" else 0.4
|
| 23 |
if 'confidence_level' not in st.session_state:
|
| 24 |
self.set_slider_value(text="Select minimum detection confidence level",
|
| 25 |
min_value=0.1,
|
|
|
|
| 19 |
if 'kbvqa' not in st.session_state:
|
| 20 |
st.session_state['kbvqa'] = None
|
| 21 |
|
| 22 |
+
self.default_confidence = 0.2 if self.detection_model == "yolov5" else 0.4
|
| 23 |
if 'confidence_level' not in st.session_state:
|
| 24 |
self.set_slider_value(text="Select minimum detection confidence level",
|
| 25 |
min_value=0.1,
|