Update README.md
Browse files
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 |
|