Spaces:

ovi054
/

image-to-prompt

Running

ovi054 commited on Jun 24

Commit

fa78611

verified ·

1 Parent(s): 2bf3775

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -37,7 +37,7 @@ def generate_caption(image):
         task="<MORE_DETAILED_CAPTION>",
         image_size=(image.width, image.height)
     )
-    prompt =  parsed_answer["<CAPTION>"]
     print("\n\nGeneration completed!:"+ prompt)
     return prompt
     # yield prompt, None

         task="<MORE_DETAILED_CAPTION>",
         image_size=(image.width, image.height)
     )
+    prompt =  parsed_answer["<MORE_DETAILED_CAPTION>"]
     print("\n\nGeneration completed!:"+ prompt)
     return prompt
     # yield prompt, None