nyust-eb210 commited on
Commit
f262453
1 Parent(s): 212eaa5

update model name

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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("bert_drcd_384")
38
- model = BertForQuestionAnswering.from_pretrained("bert_drcd_384").to(device)
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