vladyur commited on
Commit
f017bc7
·
1 Parent(s): 4f590f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, lengt
30
  eos_token_id=tokenizer.eos_token_id
31
  )
32
 
33
- return generated = list(map(tokenizer.decode, out))[0]
34
 
35
 
36
  model, tokenizer = get_model('facebook/opt-13b')
 
30
  eos_token_id=tokenizer.eos_token_id
31
  )
32
 
33
+ return list(map(tokenizer.decode, out))[0]
34
 
35
 
36
  model, tokenizer = get_model('facebook/opt-13b')