|
|
|
|
|
Fine-tuned BioGPT for Biomedical Boolean Query Formalization using Titles only. |
|
|
|
|
|
|
|
|
- Base Model: BioGPT |
|
|
- Fine-tuned on: CLEF TAR, FASS datasets |
|
|
- Task: Boolean Query Generation for PubMed searches |
|
|
|
|
|
|
|
|
```python |
|
|
from transformers import AutoModelForCausalLM, AutoTokenizer |
|
|
|
|
|
model = AutoModelForCausalLM.from_pretrained("your_username/BioGPT-BQF-Title-Small") |
|
|
tokenizer = AutoTokenizer.from_pretrained("your_username/BioGPT-BQF-Title-Small") |
|
|
|
|
|
input_text = "Best treatments for lung cancer" |
|
|
inputs = tokenizer(input_text, return_tensors="pt") |
|
|
outputs = model.generate(**inputs) |
|
|
print(tokenizer.decode(outputs[0], skip_special_tokens=True)) |
|
|
|
