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
|