tahsinhasem commited on
Commit
b40fa52
·
verified ·
1 Parent(s): 8b05f55

Respond with Special Tokens

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -80,7 +80,7 @@ async def generate_text(item: Item):
80
  pad_token_id=tokenizer.eos_token_id # Set this to suppress warning
81
  )
82
 
83
- resp = tokenizer.decode(outputs[0], skip_special_tokens=True)
84
 
85
 
86
  return {"response": resp}
 
80
  pad_token_id=tokenizer.eos_token_id # Set this to suppress warning
81
  )
82
 
83
+ resp = tokenizer.decode(outputs[0], skip_special_tokens=False)
84
 
85
 
86
  return {"response": resp}