Spaces:
Sleeping
Sleeping
Update app.py
Browse filesUpdated line 25 code
app.py
CHANGED
|
@@ -22,7 +22,8 @@ story_generator = pipeline(
|
|
| 22 |
# Function to generate stories
|
| 23 |
def generate_story(prompt, max_length=300, temperature=0.8):
|
| 24 |
outputs = story_generator(
|
| 25 |
-
|
|
|
|
| 26 |
max_length=max_length,
|
| 27 |
temperature=temperature,
|
| 28 |
do_sample=True,
|
|
|
|
| 22 |
# Function to generate stories
|
| 23 |
def generate_story(prompt, max_length=300, temperature=0.8):
|
| 24 |
outputs = story_generator(
|
| 25 |
+
model,
|
| 26 |
+
#prompt,
|
| 27 |
max_length=max_length,
|
| 28 |
temperature=temperature,
|
| 29 |
do_sample=True,
|