Spaces:
Sleeping
Sleeping
Ezi Ozoani
commited on
Commit
·
b2e2fd7
1
Parent(s):
516fb17
spelling tweak
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def load_model():
|
|
35 |
text = st.text_input('Provide an initial text prompt')
|
36 |
|
37 |
if text != '' :
|
38 |
-
out = generator(text, max_length=
|
39 |
|
40 |
def cs_sidebar():
|
41 |
|
@@ -285,7 +285,7 @@ output = model(encoded_input)
|
|
285 |
|
286 |
# Try App
|
287 |
|
288 |
-
col2.header('Try out
|
289 |
#print load_model()
|
290 |
with col2.subheader(''):
|
291 |
generator = pipeline('text-generation', model='distilgpt2')
|
|
|
35 |
text = st.text_input('Provide an initial text prompt')
|
36 |
|
37 |
if text != '' :
|
38 |
+
out = generator(text, max_length=0, num_return_sequences=1)
|
39 |
|
40 |
def cs_sidebar():
|
41 |
|
|
|
285 |
|
286 |
# Try App
|
287 |
|
288 |
+
col2.header('Try out DistilGPT2')
|
289 |
#print load_model()
|
290 |
with col2.subheader(''):
|
291 |
generator = pipeline('text-generation', model='distilgpt2')
|