Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
@@ -98,11 +98,11 @@ def run_inference():
|
|
98 |
st.title("Run Inference")
|
99 |
|
100 |
|
101 |
-
st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
|
102 |
-
st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
|
103 |
default_confidence = 0.2 if st.session_state['detection_model'] == "yolov5" else 0.4
|
104 |
# confidence_level = st.slider("Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.1)
|
105 |
-
state_manager.set_slider_value(text="Select minimum detection confidence level",
|
106 |
min_value=0.1,
|
107 |
max_value=0.9,
|
108 |
value=default_confidence,
|
|
|
98 |
st.title("Run Inference")
|
99 |
|
100 |
|
101 |
+
method = st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
|
102 |
+
detection_model = st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
|
103 |
default_confidence = 0.2 if st.session_state['detection_model'] == "yolov5" else 0.4
|
104 |
# confidence_level = st.slider("Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.1)
|
105 |
+
confidence_level = state_manager.set_slider_value(text="Select minimum detection confidence level",
|
106 |
min_value=0.1,
|
107 |
max_value=0.9,
|
108 |
value=default_confidence,
|