Spaces:

autonomous019
/

image_story_generator

Runtime error

autonomous019 commited on Aug 31, 2022

Commit

9f74712

1 Parent(s): 999d49b

bug fixing

Files changed (1) hide show

app.py CHANGED Viewed

@@ -71,7 +71,7 @@ def self_caption(image):
   #for prediction in preds:
   #  pred_output[predicted_label] = prediction
-  return pred_output
 def classify_image(image):
   results = image_pipe(image)

   #for prediction in preds:
   #  pred_output[predicted_label] = prediction
+  return preds[0]
 def classify_image(image):
   results = image_pipe(image)