Update app.py

app.py

@@ -54,10 +54,11 @@ def inference(img):
     # for name, prob in zip(pred_class_names, pred_class_probs):
     #     res += f"[{prob:2.2f}%]\t{name}\n"
 
-    return {imagenet_id_to_classname[str(i)]: float(prediction[i]) for i in range(1000)}
+    return {imagenet_id_to_classname[str(i.item())]: float(prediction[i]) for i in range(1000)}
     
 inputs = gr.inputs.Image(type='pil')
-outputs = gr.outputs.Textbox(label="Output")
+# outputs = gr.outputs.Textbox(label="Output")
+label = gr.outputs.Label(num_top_classes=5)
 
 title = "UniFormer-S"
 
@@ -66,4 +67,4 @@ description = "Gradio demo for UniFormer: To use it, simply upload your image, o
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2201.09450' target='_blank'>UniFormer: Unifying Convolution and Self-attention for Visual Recognition</a> | <a href='https://github.com/Sense-X/UniFormer' target='_blank'>Github Repo</a></p>"
 
 
-gr.Interface(inference, inputs, outputs, title=title, description=description, article=article, examples=[['library.jpg']]).launch(enable_queue=True,cache_examples=True)
+gr.Interface(inference, inputs, outputs=label, title=title, description=description, article=article, examples=[['library.jpg']]).launch(enable_queue=True,cache_examples=True)