Pippoz commited on
Commit
a95e7a5
·
1 Parent(s): ada482c

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)
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=1)
29
 
30
  #answer = opt_model(prompt, model, tokenizer,)
31
  #lst = ['ciao come stai sjfsbd dfhsdf fuahfuf feuhfu wefwu ']