Update app.py

app.py

@@ -212,7 +212,7 @@ def inference(img):
   for i, item in enumerate(top_5):
     class_index = item if includes_background_class else item + 1
     line = f'({i+1}) {class_index:4} - {classes[class_index]}: {probabilities[0][top_5][i]}'
-    result[classes[class_index]] = probabilities[0][top_5][i]
+    result[classes[class_index]] = probabilities[0][top_5][i].item()
   return result
   
 gr.Interface(inference,gr.inputs.Image(type="filepath"),"label").launch()