Update app.py
Browse files
app.py
CHANGED
|
@@ -38,7 +38,7 @@ def generate_content(image):
|
|
| 38 |
# Generate raw story
|
| 39 |
raw = storyer(
|
| 40 |
prompt,
|
| 41 |
-
max_new_tokens=
|
| 42 |
temperature=0.7,
|
| 43 |
top_p=0.9,
|
| 44 |
no_repeat_ngram_size=2,
|
|
|
|
| 38 |
# Generate raw story
|
| 39 |
raw = storyer(
|
| 40 |
prompt,
|
| 41 |
+
max_new_tokens=200,
|
| 42 |
temperature=0.7,
|
| 43 |
top_p=0.9,
|
| 44 |
no_repeat_ngram_size=2,
|