phamson02
commited on
Commit
•
06071ca
1
Parent(s):
d0547af
update
Browse files- generate_poem.py +1 -1
generate_poem.py
CHANGED
|
@@ -13,7 +13,7 @@ def generate_poem(input_text):
|
|
| 13 |
|
| 14 |
# Define the parameters for the generate function
|
| 15 |
min_length = 50
|
| 16 |
-
max_length =
|
| 17 |
rep_penalty = 1.2
|
| 18 |
temp = 0.7
|
| 19 |
top_k = 50
|
|
|
|
| 13 |
|
| 14 |
# Define the parameters for the generate function
|
| 15 |
min_length = 50
|
| 16 |
+
max_length = 512
|
| 17 |
rep_penalty = 1.2
|
| 18 |
temp = 0.7
|
| 19 |
top_k = 50
|