Commit
•
606792c
1
Parent(s):
914343a
Update README.md
Browse files
README.md
CHANGED
|
@@ -55,7 +55,9 @@ res = nlp(qa_input)
|
|
| 55 |
# {'score': 0.980, 'start': 30, 'end': 37, 'answer': ' London'}
|
| 56 |
|
| 57 |
# b) Load model & tokenizer
|
| 58 |
-
model = AutoModelForQuestionAnswering.from_pretrained(
|
|
|
|
|
|
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 60 |
|
| 61 |
question = f'{tokenizer.cls_token}Where do I live?' # '<cls>Where do I live?'
|
|
|
|
| 55 |
# {'score': 0.980, 'start': 30, 'end': 37, 'answer': ' London'}
|
| 56 |
|
| 57 |
# b) Load model & tokenizer
|
| 58 |
+
model = AutoModelForQuestionAnswering.from_pretrained(
|
| 59 |
+
model_name, trust_remote_code=True
|
| 60 |
+
)
|
| 61 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 62 |
|
| 63 |
question = f'{tokenizer.cls_token}Where do I live?' # '<cls>Where do I live?'
|