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