Update app.py

app.py

@@ -12,7 +12,7 @@ choices = gr.inputs.Dropdown(choices=models, default='flyswot/convnext-tiny-224_
 
 def predict(image,model):
     pipe = pipeline("image-classification", model=model)
-    predictions = pipe(image)
+    predictions = pipe(image, top_k=20)
     return {pred['label']: pred['score'] for pred in predictions}
 
 iface = gr.Interface(