mrm8488 commited on
Commit
2e333f5
1 Parent(s): 6bcb233

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -63,7 +63,7 @@ def get_response(question, context, max_length=32):
63
  attention_mask=features['attention_mask'].to(device),
64
  max_length=max_length)
65
 
66
- return tokenizer.decode(output[0])
67
 
68
  # Some examples in different languages
69
 
63
  attention_mask=features['attention_mask'].to(device),
64
  max_length=max_length)
65
 
66
+ return tokenizer.decode(output[0], skip_special_tokens=True)
67
 
68
  # Some examples in different languages
69