nyust-eb210
commited on
Commit
·
f262453
1
Parent(s):
212eaa5
update model name
Browse files
README.md
CHANGED
@@ -34,8 +34,8 @@ Training Arguments:
|
|
34 |
text = "馬雲是我爸爸。"
|
35 |
query = "我爸爸是誰?"
|
36 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
37 |
-
tokenizer = BertTokenizerFast.from_pretrained("
|
38 |
-
model = BertForQuestionAnswering.from_pretrained("
|
39 |
encoded_input = tokenizer(text, query, return_tensors="pt").to(device)
|
40 |
qa_outputs = model(**encoded_input)
|
41 |
|
|
|
34 |
text = "馬雲是我爸爸。"
|
35 |
query = "我爸爸是誰?"
|
36 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
37 |
+
tokenizer = BertTokenizerFast.from_pretrained("nyust-eb210/braslab-bert-drcd-384")
|
38 |
+
model = BertForQuestionAnswering.from_pretrained("nyust-eb210/braslab-bert-drcd-384").to(device)
|
39 |
encoded_input = tokenizer(text, query, return_tensors="pt").to(device)
|
40 |
qa_outputs = model(**encoded_input)
|
41 |
|