Spaces:
Sleeping
Sleeping
Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
@@ -99,7 +99,7 @@ def run_inference():
|
|
99 |
state_manager.display_session_state()
|
100 |
st.title("Run Inference")
|
101 |
|
102 |
-
|
103 |
method = st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
|
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
|
@@ -109,7 +109,7 @@ def run_inference():
|
|
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()
|
|
|
99 |
state_manager.display_session_state()
|
100 |
st.title("Run Inference")
|
101 |
|
102 |
+
|
103 |
method = st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
|
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
|
|
|
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()
|