Spaces:
Runtime error
Runtime error
Commit
·
1556d5c
1
Parent(s):
8d609d2
Update app.py
Browse files
app.py
CHANGED
@@ -7,14 +7,16 @@ st.markdown('## Text-generation gpt Muse from Breadlicker45')
|
|
7 |
use_auth_token=True
|
8 |
@st.cache(allow_output_mutation=True, suppress_st_warning =True, show_spinner=False)
|
9 |
def get_model():
|
10 |
-
return pipeline('text-generation', model=model, do_sample=
|
11 |
|
12 |
col1, col2 = st.columns([2,1])
|
13 |
|
14 |
with st.sidebar:
|
15 |
st.markdown('## Model Parameters')
|
16 |
|
17 |
-
max_length = st.slider('Max text length',
|
|
|
|
|
18 |
|
19 |
num_beams = st.slider('N° tree beams search', 1, 15, 1)
|
20 |
|
@@ -54,7 +56,7 @@ with col2:
|
|
54 |
gen = st.info('Generating text...')
|
55 |
answer = generator(prompt,
|
56 |
max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
|
57 |
-
early_stopping=early_stopping, num_beams=num_beams,
|
58 |
gen.empty()
|
59 |
|
60 |
lst = answer[0]['generated_text']
|
|
|
7 |
use_auth_token=True
|
8 |
@st.cache(allow_output_mutation=True, suppress_st_warning =True, show_spinner=False)
|
9 |
def get_model():
|
10 |
+
return pipeline('text-generation', model=model, do_sample=True)
|
11 |
|
12 |
col1, col2 = st.columns([2,1])
|
13 |
|
14 |
with st.sidebar:
|
15 |
st.markdown('## Model Parameters')
|
16 |
|
17 |
+
max_length = st.slider('Max text length', 80, 2000, 80)
|
18 |
+
|
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 |
|
|
|
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']
|