Update app.py

app.py

@@ -31,7 +31,7 @@ def predict(image):
     predicted_class_idx = logits.argmax(-1).item()
     return(model.config.id2label[predicted_class_idx])
 
-demo = gr.Interface(fn=predict, inputs="image", outputs="text", title=title, examples=["examples/image_1.png", "examples/image_2.png", "examples/image_3.png"],
+demo = gr.Interface(fn=predict, inputs="image", outputs=gr.Label(num_top_classes=2), title=title, examples=["examples/image_1.png", "examples/image_2.png", "examples/image_3.png"],
     description=description,).launch()
 
 # demo.launch(debug=True)