Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ def text2story(text):
|
|
| 14 |
The story must contain:
|
| 15 |
- Beginning, middle, end
|
| 16 |
- Character development"""
|
| 17 |
-
text_to_story_model= pipeline("text-generation", model="
|
| 18 |
story_text = text_to_story_model(
|
| 19 |
messages,
|
| 20 |
max_length=200,
|
|
|
|
| 14 |
The story must contain:
|
| 15 |
- Beginning, middle, end
|
| 16 |
- Character development"""
|
| 17 |
+
text_to_story_model= pipeline("text-generation", model="openai-community/gpt2")
|
| 18 |
story_text = text_to_story_model(
|
| 19 |
messages,
|
| 20 |
max_length=200,
|