fix: predict_fn return bug

app.py

@@ -38,6 +38,6 @@ def predict_fn(image):
         top_p, top_class = probabilities.topk(1, dim=1)
 
         # return dictionary whose keys are labels and values are confidences
-        return {str(top_class.item()), str(top_p.item())}
+        return {str(top_class.item()): str(top_p.item())}
 
 gr.Interface(predict_fn, gr.inputs.Image(type="pil", label="Input Image"), outputs="label").launch()