m7mdal7aj commited on
Commit
f6f453e
1 Parent(s): 846f1f8

Update my_model/tabs/run_inference.py

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