Spaces:
Paused
Paused
Update app.py
Browse files
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)
|