Pippoz commited on
Commit
3820aa0
1 Parent(s): 660ebbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ generator = pipeline('text-generation', model="facebook/opt-1.3b", skip_special_
25
 
26
  prompt= st.text_area('Your prompt here',
27
  '''Hello, I'm am conscious and''')
28
- answer = generator(prompt, max_length=60,no_repeat_ngram_size=1)
29
 
30
  #answer = opt_model(prompt, model, tokenizer,)
31
  #lst = ['ciao come stai sjfsbd dfhsdf fuahfuf feuhfu wefwu ']
 
25
 
26
  prompt= st.text_area('Your prompt here',
27
  '''Hello, I'm am conscious and''')
28
+ answer = generator(prompt, max_length=60,no_repeat_ngram_size=2, early_stopping=True, num_beams=5)
29
 
30
  #answer = opt_model(prompt, model, tokenizer,)
31
  #lst = ['ciao come stai sjfsbd dfhsdf fuahfuf feuhfu wefwu ']