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