KantaHayashiAI commited on
Commit
b028eac
·
verified ·
1 Parent(s): 6308af3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def generate(
62
  )
63
 
64
  generated_segment = output_ids[0][input_ids.shape[1]:]
65
- return tokenizer.decode(generated_segment, skip_special_tokens=False, clean_up_tokenization_spaces=False)
66
 
67
 
68
  demo = gr.ChatInterface(
 
62
  )
63
 
64
  generated_segment = output_ids[0][input_ids.shape[1]:]
65
+ return tokenizer.decode(generated_segment, skip_special_tokens=True)
66
 
67
 
68
  demo = gr.ChatInterface(