m7mdal7aj commited on
Commit
198c679
·
verified ·
1 Parent(s): 105e89e

Update my_model/tabs/run_inference.py

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