Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
23 |
-
model = BioGptForCausalLM.from_pretrained(
|
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 |
|