sagemaker commited on
Commit
2b7f82a
1 Parent(s): 42d774c

updates labels

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -79,9 +79,9 @@ Leaving the field empty will show all classes"""
79
 
80
  image_in = gr.components.Image()
81
  image_out = gr.components.Image()
82
- model_choice = gr.components.Dropdown(["yolos-tiny", "yolos-small", "yolos_base", "yolos-small-300", "yolos-small-dwr"], value="yolos-small")
83
  prob_threshold_slider = gr.components.Slider(minimum=0, maximum=1.0, step=0.01, value=0.9, label="Probability Threshold")
84
- classes_to_show = gr.components.Textbox(placeholder="e.g. person, boat")
85
 
86
  Iface = gr.Interface(
87
  fn=infer,
 
79
 
80
  image_in = gr.components.Image()
81
  image_out = gr.components.Image()
82
+ model_choice = gr.components.Dropdown(["yolos-tiny", "yolos-small", "yolos_base", "yolos-small-300", "yolos-small-dwr"], value="yolos-small", label="YOLOS Model")
83
  prob_threshold_slider = gr.components.Slider(minimum=0, maximum=1.0, step=0.01, value=0.9, label="Probability Threshold")
84
+ classes_to_show = gr.components.Textbox(placeholder="e.g. person, boat", label="Classes to use (empty means all classes)")
85
 
86
  Iface = gr.Interface(
87
  fn=infer,