m7mdal7aj commited on
Commit
e79add6
1 Parent(s): d163e0e

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +3 -3
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,