bugs

app.py

@@ -71,8 +71,14 @@ def self_caption(image):
   print("Predictions")
   print(preds)
   print("The preds type is : ",type(preds))
-  
-  return(preds[0].split('.')[0])
+  pred_keys = ["Prediction"]
+  pred_value = preds
+
+  pred_dictionary = dict(zip(pred_keys, pred_value))
+  print("Pred dictionary")
+  print(pred_dictionary)
+  return(pred_dictionary)
+  #return(preds[0].split('.')[0])
 
 def classify_image(image):
   results = image_pipe(image)