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,
|