| |
| 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)) |
|
|
