Spaces:
Runtime error
Runtime error
Commit
•
37ae943
1
Parent(s):
1556d5c
Update app.py
Browse files
app.py
CHANGED
@@ -19,6 +19,8 @@ with st.sidebar:
|
|
19 |
min_length = st.slider('Min text length', 80, 500, 80)
|
20 |
|
21 |
num_beams = st.slider('N° tree beams search', 1, 15, 1)
|
|
|
|
|
22 |
|
23 |
early_stopping = st.selectbox(
|
24 |
'Early stopping text generation',
|
@@ -56,7 +58,7 @@ with col2:
|
|
56 |
gen = st.info('Generating text...')
|
57 |
answer = generator(prompt,
|
58 |
max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
|
59 |
-
early_stopping=early_stopping, num_beams=num_beams, min_length=min_length)
|
60 |
gen.empty()
|
61 |
|
62 |
lst = answer[0]['generated_text']
|
|
|
19 |
min_length = st.slider('Min text length', 80, 500, 80)
|
20 |
|
21 |
num_beams = st.slider('N° tree beams search', 1, 15, 1)
|
22 |
+
|
23 |
+
temp = st.slider('temperature', 0, 1, 1)
|
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, temperature=temp)
|
62 |
gen.empty()
|
63 |
|
64 |
lst = answer[0]['generated_text']
|