Spaces:
Running
Running
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')
|