fix bug

app.py

@@ -23,11 +23,8 @@ def predict_fn(img):
     with torch.no_grad():
         out = model(img)
 
-    probalilities = torch.nn.functional.softmax(out[0], dim=0)
-
-    values, indices = torch.topk(probalilities, k=5)
-
-    return [LABELS[i]: v.item() for i, v in zip(indices, values)]
-gr.Interface(predict_fn, gr.inputs.Image(type='pil'), outputs='label').launch()
+    probabilities = torch.nn.functional.softmax(out[0], dim=0)
 
+    values, indices = torch.topk(probabilities, k=5)
 
+    return [{LABELS[i]: v.item()} for i, v in zip(indices, values)]