juliensimon HF staff commited on
Commit
bc9d9eb
1 Parent(s): 2eb5044

5 labels max

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=10, 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
  )
 
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
  )