Fix ouput dict containing tensor instead of float

app.py

@@ -7,7 +7,7 @@ def classify_image(img):
     celebrity_name,_,probs = learn.predict(img)
     results = dict(zip(list(learn.dls.vocab),probs))
     best_results = sorted(results, key=results.get, reverse=True)[:5]
-    probabilities = {key: results[key] for key in best_results}
+    probabilities = {key: float(results[key]) for key in best_results}
     return {probabilities}
 
 image = gr.inputs.Image(shape=(224,224))