storresbusquets commited on
Commit
ca29b6a
·
1 Parent(s): b5e77db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ class GradioInference:
57
  do_sample=True,
58
  top_k=10,
59
  num_return_sequences=1,
60
- eos_token_id=tokenizer.eos_token_id
61
  )
62
 
63
 
 
57
  do_sample=True,
58
  top_k=10,
59
  num_return_sequences=1,
60
+ eos_token_id=self.tokenizer.eos_token_id
61
  )
62
 
63