Update README.md
Browse files
README.md
CHANGED
@@ -39,8 +39,8 @@ Fine-tuned BioGPT for Biomedical Boolean Query Formalization using Titles only.
|
|
39 |
```python
|
40 |
from transformers import BioGptForCausalLM, BioGptTokenizer
|
41 |
|
42 |
-
model = BioGptForCausalLM.from_pretrained("
|
43 |
-
tokenizer = BioGptTokenizer.from_pretrained("
|
44 |
|
45 |
input_text = "Title: Heterogeneity in Lung Cancer, Query: "
|
46 |
inputs = tokenizer(input_text, return_tensors="pt")
|
|
|
39 |
```python
|
40 |
from transformers import BioGptForCausalLM, BioGptTokenizer
|
41 |
|
42 |
+
model = BioGptForCausalLM.from_pretrained("AI4BSLR/BioGPT-BQF-Title-Small")
|
43 |
+
tokenizer = BioGptTokenizer.from_pretrained("AI4BSLR/BioGPT-BQF-Title-Small")
|
44 |
|
45 |
input_text = "Title: Heterogeneity in Lung Cancer, Query: "
|
46 |
inputs = tokenizer(input_text, return_tensors="pt")
|