Spaces:

flyswot
/

model_explorer

Sleeping

davanstrien HF Staff commited on May 5, 2022

Commit

0114d4f

1 Parent(s): 111f215

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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)
     return {pred['label']: pred['score'] for pred in predictions}
 iface = gr.Interface(

 def predict(image,model):
     pipe = pipeline("image-classification", model=model)
+    predictions = pipe(image, top_k=20)
     return {pred['label']: pred['score'] for pred in predictions}
 iface = gr.Interface(