Update app.py

app.py

@@ -14,7 +14,11 @@ for model_id in get_model_ids():
   models[model_id] = pipeline("image-classification", model=model_id)
     
 def predict(img, model_id):
-  return models[model_id](img)
+  preds =  models[model_id](img)
+  res = {}
+  for pred in preds:
+    res[pred["label"]] = pred["score"]
+  return res
   
 gr.Interface(
   fn=predict,