Spaces:

Manikandan-Alagu
/

Auto-Poem-Generator

Sleeping

MANIKANDAN A commited on Jul 19, 2023

Commit

443cdfd

1 Parent(s): 2329ed9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -369,7 +369,7 @@ def predict(term_col, poem_col):
     pred_caption = generate_caption('tmp.jpg', caption_model)
     # Generate poem using poetpy
-    poem_lines = generate_poem(pred_caption, num_lines=10)
     # Display the poem
     poem_col.markdown('#### Generated Poem:')

     pred_caption = generate_caption('tmp.jpg', caption_model)
     # Generate poem using poetpy
+    poem_lines = generate_poem(pred_caption, 10)
     # Display the poem
     poem_col.markdown('#### Generated Poem:')