maliozer commited on
Commit
13909fd
1 Parent(s): 8787192

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ def load_model(model_name="BioGPT"):
19
  "BioGPT-QA-PubMedQA-BioGPT":"microsoft/BioGPT-Large-PubMedQA"
20
  }
21
 
22
- tokenizer = BioGptTokenizer.from_pretrained(model_name_map[model_name])
23
- model = BioGptForCausalLM.from_pretrained(model_name_map[model_name])
24
  return tokenizer, model
25
 
26
 
 
19
  "BioGPT-QA-PubMedQA-BioGPT":"microsoft/BioGPT-Large-PubMedQA"
20
  }
21
 
22
+ tokenizer = BioGptTokenizer.from_pretrained("microsoft/biogpt")
23
+ model = BioGptForCausalLM.from_pretrained("microsoft/biogpt")
24
  return tokenizer, model
25
 
26