Update app.py

app.py

@@ -208,11 +208,11 @@ def inference(img):
   # Some models include an additional 'background' class in the predictions, so
   # we must account for this when reading the class labels.
   includes_background_class = probabilities.shape[1] == 1001
-  
+  result = {}
   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]] = result[probabilities[0][top_5][i]]
+  return result
   
-  return line
-  
-gr.Interface(inference,gr.inputs.Image(type="filepath"),"text").launch()
+gr.Interface(inference,gr.inputs.Image(type="filepath"),"label").launch()