Spaces:
Build error
Build error
sagemaker
commited on
Commit
•
a57a57c
1
Parent(s):
2b7f82a
fix typo
Browse files
app.py
CHANGED
@@ -79,7 +79,7 @@ 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", "
|
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 |
|
|
|
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 |
|