Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -47,8 +47,7 @@ with col2:
|
|
| 47 |
gen = st.info('Generating text...')
|
| 48 |
answer = generator(prompt,
|
| 49 |
max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
|
| 50 |
-
early_stopping=early_stopping, num_beams=num_beams
|
| 51 |
-
skip_special_tokens=True)
|
| 52 |
gen.empty()
|
| 53 |
|
| 54 |
lst = answer[0]['generated_text']
|
|
|
|
| 47 |
gen = st.info('Generating text...')
|
| 48 |
answer = generator(prompt,
|
| 49 |
max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
|
| 50 |
+
early_stopping=early_stopping, num_beams=num_beams)
|
|
|
|
| 51 |
gen.empty()
|
| 52 |
|
| 53 |
lst = answer[0]['generated_text']
|