Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
@@ -100,17 +100,7 @@ def run_inference():
|
|
100 |
st.title("Run Inference")
|
101 |
|
102 |
|
103 |
-
|
104 |
-
detection_model = st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
|
105 |
-
default_confidence = 0.2 if detection_model == "yolov5" else 0.4
|
106 |
-
# confidence_level = st.slider("Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.1)
|
107 |
-
confidence_level = state_manager.set_slider_value(text="Select minimum detection confidence level",
|
108 |
-
min_value=0.1,
|
109 |
-
max_value=0.9,
|
110 |
-
value=default_confidence,
|
111 |
-
step=0.1,
|
112 |
-
slider_key_name='confidence_level'
|
113 |
-
)
|
114 |
|
115 |
state_manager.display_model_settings()
|
116 |
# state_manager.update_model_settings(detection_model=detection_model, confidence_level=confidence_level, selected_method=method)
|
|
|
100 |
st.title("Run Inference")
|
101 |
|
102 |
|
103 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
104 |
|
105 |
state_manager.display_model_settings()
|
106 |
# state_manager.update_model_settings(detection_model=detection_model, confidence_level=confidence_level, selected_method=method)
|