Ezi Ozoani commited on
Commit
b2e2fd7
1 Parent(s): 516fb17

spelling tweak

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=30, num_return_sequences=1)
39
 
40
  def cs_sidebar():
41
 
@@ -285,7 +285,7 @@ output = model(encoded_input)
285
 
286
  # Try App
287
 
288
- col2.header('Try out DistilGP2')
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')