Spaces:
Runtime error
Runtime error
Commit
•
69fb956
1
Parent(s):
0d73f08
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ with st.sidebar:
|
|
16 |
|
17 |
max_length = st.slider('Max text length', 0, 2000, 80)
|
18 |
|
19 |
-
num_beams = st.slider('N° tree beams search',
|
20 |
|
21 |
early_stopping = st.selectbox(
|
22 |
'Early stopping text generation',
|
@@ -48,7 +48,7 @@ with col2:
|
|
48 |
|
49 |
gen = st.info('Generating text...')
|
50 |
answer = generator(prompt,
|
51 |
-
max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
|
52 |
early_stopping=early_stopping, num_beams=num_beams)
|
53 |
gen.empty()
|
54 |
|
|
|
16 |
|
17 |
max_length = st.slider('Max text length', 0, 2000, 80)
|
18 |
|
19 |
+
num_beams = st.slider('N° tree beams search', 10, 150, 15)
|
20 |
|
21 |
early_stopping = st.selectbox(
|
22 |
'Early stopping text generation',
|
|
|
48 |
|
49 |
gen = st.info('Generating text...')
|
50 |
answer = generator(prompt,
|
51 |
+
max_length=max_length, no_repeat_ngram_size=(no_ngram_repeat/10),
|
52 |
early_stopping=early_stopping, num_beams=num_beams)
|
53 |
gen.empty()
|
54 |
|