m7mdal7aj commited on
Commit
31da5e4
1 Parent(s): 19fb14c

Update my_model/state_manager.py

Browse files
Files changed (1) hide show
  1. my_model/state_manager.py +1 -1
my_model/state_manager.py CHANGED
@@ -64,7 +64,7 @@ class StateManager:
64
  Sets up user interface widgets for selecting models, settings, and displaying model settings conditionally.
65
  """
66
 
67
- self.col1.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)", "Learning Visual Embeddings"], index=0, key='method', disabled=self.is_widget_disabled)
68
  detection_model = self.col1.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model', disabled=self.is_widget_disabled)
69
  default_confidence = 0.2 if st.session_state.detection_model == "yolov5" else 0.4
70
  self.set_slider_value(text="Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.05, slider_key_name='confidence_level', col=self.col1)
 
64
  Sets up user interface widgets for selecting models, settings, and displaying model settings conditionally.
65
  """
66
 
67
+ self.col1.selectbox("Choose a method:", ["13b-Fine-Tuned Model", "7b-Fine-Tuned Model", "Learning Visual Embeddings"], index=0, key='method', disabled=self.is_widget_disabled)
68
  detection_model = self.col1.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model', disabled=self.is_widget_disabled)
69
  default_confidence = 0.2 if st.session_state.detection_model == "yolov5" else 0.4
70
  self.set_slider_value(text="Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.05, slider_key_name='confidence_level', col=self.col1)