Spaces:
Runtime error
Runtime error
Julien Simon
commited on
Commit
·
bc9d9eb
1
Parent(s):
2eb5044
5 labels max
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def process(image_file, top_k, model_name):
|
|
22 |
|
23 |
# Inputs
|
24 |
image = gr.Image(type="filepath", label="Upload an image")
|
25 |
-
top_k = gr.Slider(minimum=1, maximum=
|
26 |
model_selection = gr.Dropdown(
|
27 |
model_names, value="google/vit-base-patch16-224", label="Pick a model"
|
28 |
)
|
|
|
22 |
|
23 |
# Inputs
|
24 |
image = gr.Image(type="filepath", label="Upload an image")
|
25 |
+
top_k = gr.Slider(minimum=1, maximum=5, step=1, value=5, label="Top k classes")
|
26 |
model_selection = gr.Dropdown(
|
27 |
model_names, value="google/vit-base-patch16-224", label="Pick a model"
|
28 |
)
|