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 |
|