Spaces:
Build error
Build error
sagemaker
commited on
Commit
•
2b7f82a
1
Parent(s):
42d774c
updates labels
Browse files
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,
|