Spaces:
Runtime error
Runtime error
Commit
•
650f659
1
Parent(s):
fe8a926
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ with st.sidebar:
|
|
20 |
|
21 |
num_beams = st.slider('N° tree beams search', 1, 15, 1)
|
22 |
|
23 |
-
|
24 |
|
25 |
early_stopping = st.selectbox(
|
26 |
'Early stopping text generation',
|
@@ -58,7 +58,7 @@ with col2:
|
|
58 |
gen = st.info('Generating text...')
|
59 |
answer = generator(prompt,
|
60 |
max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
|
61 |
-
early_stopping=early_stopping, num_beams=num_beams, min_length=min_length
|
62 |
gen.empty()
|
63 |
|
64 |
lst = answer[0]['generated_text']
|
|
|
20 |
|
21 |
num_beams = st.slider('N° tree beams search', 1, 15, 1)
|
22 |
|
23 |
+
|
24 |
|
25 |
early_stopping = st.selectbox(
|
26 |
'Early stopping text generation',
|
|
|
58 |
gen = st.info('Generating text...')
|
59 |
answer = generator(prompt,
|
60 |
max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
|
61 |
+
early_stopping=early_stopping, num_beams=num_beams, min_length=min_length)
|
62 |
gen.empty()
|
63 |
|
64 |
lst = answer[0]['generated_text']
|