Update app.py
Browse files
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,
|
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:
|