Spaces:
Runtime error
Runtime error
Update pages/text_gen.py
Browse files- pages/text_gen.py +2 -2
pages/text_gen.py
CHANGED
@@ -37,8 +37,8 @@ def app():
|
|
37 |
with col1:
|
38 |
gen = st.info('Generating text...')
|
39 |
answer = generator(prompt,
|
40 |
-
max_length=
|
41 |
-
early_stopping=
|
42 |
skip_special_tokens=True)
|
43 |
gen.empty()
|
44 |
|
|
|
37 |
with col1:
|
38 |
gen = st.info('Generating text...')
|
39 |
answer = generator(prompt,
|
40 |
+
max_length=80, no_repeat_ngram_size=2,
|
41 |
+
early_stopping=True, num_beams=8,
|
42 |
skip_special_tokens=True)
|
43 |
gen.empty()
|
44 |
|