heisenberg3376 commited on
Commit
ca3756e
·
verified ·
1 Parent(s): 10f8736

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def classify_image(input_image):
21
  iface = gr.Interface(
22
  fn=classify_image,
23
  inputs=gr.Image(type="pil", label="Upload an image"),
24
- outputs="text",
25
  title="Image Classification",
26
  description="Classify food items in images using heisenberg3376/vit-base-food-items-v1"
27
  )
 
21
  iface = gr.Interface(
22
  fn=classify_image,
23
  inputs=gr.Image(type="pil", label="Upload an image"),
24
+ outputs=gr.Label(num_top_classes=5, label="Top 5 Predictions"),
25
  title="Image Classification",
26
  description="Classify food items in images using heisenberg3376/vit-base-food-items-v1"
27
  )