bug tracing

app.py

@@ -64,8 +64,11 @@ def self_caption(image):
   preds = [pred.strip() for pred in preds]
   print("Predictions")
   print(preds)
-  preds = ' '.join(preds)
-  return preds
+  pred_output = {}
+  for prediction in preds:
+    pred_output[predicted_label] = prediction
+  
+  return pred_output
 
 def classify_image(image):
   results = image_pipe(image)