MANIKANDAN A commited on
Commit
443cdfd
1 Parent(s): 2329ed9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -369,7 +369,7 @@ def predict(term_col, poem_col):
369
  pred_caption = generate_caption('tmp.jpg', caption_model)
370
 
371
  # Generate poem using poetpy
372
- poem_lines = generate_poem(pred_caption, num_lines=10)
373
 
374
  # Display the poem
375
  poem_col.markdown('#### Generated Poem:')
 
369
  pred_caption = generate_caption('tmp.jpg', caption_model)
370
 
371
  # Generate poem using poetpy
372
+ poem_lines = generate_poem(pred_caption, 10)
373
 
374
  # Display the poem
375
  poem_col.markdown('#### Generated Poem:')