Update app.py

app.py

@@ -54,7 +54,7 @@ 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)]: float(prediction[0][i]) for i in range(1000)}
     
 inputs = gr.inputs.Image(type='pil')
 # outputs = gr.outputs.Textbox(label="Output")