Tonic commited on
Commit
098969f
·
1 Parent(s): 51db12e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -56,6 +56,8 @@ class OrcaChatBot:
56
  repetition_penalty=repetition_penalty,
57
  pad_token_id=self.tokenizer.eos_token_id,
58
  do_sample=True
 
 
59
  )
60
 
61
  response = self.tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
56
  repetition_penalty=repetition_penalty,
57
  pad_token_id=self.tokenizer.eos_token_id,
58
  do_sample=True
59
+ skip_prompt=True,
60
+ skip_special_tokens=True
61
  )
62
 
63
  response = self.tokenizer.decode(output_ids[0], skip_special_tokens=True)