xieqilenb commited on
Commit
6c8ca8d
·
verified ·
1 Parent(s): ab58353

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def generate_story(caption):
31
  f"Image Details: {caption}\n\nStory:"
32
  )
33
 
34
- result = story_generator(prompt, max_length=300, num_return_sequences=1)
35
  full_text = result[0]['generated_text']
36
 
37
  if "Story:" in full_text:
 
31
  f"Image Details: {caption}\n\nStory:"
32
  )
33
 
34
+ result = story_generator(prompt, num_return_sequences=1)
35
  full_text = result[0]['generated_text']
36
 
37
  if "Story:" in full_text: