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()
|