Views
No views yet
1# Example code for model inference
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4tokenizer = AutoTokenizer.from_pretrained("DXCLab/OncoCareBrain-GPT")
5model = AutoModelForCausalLM.from_pretrained("DXCLab/OncoCareBrain-GPT")
6
7input_text = "Could you analyze this genomic profile and suggest potential treatment options for breast cancer with BRCA1 mutation?"
8inputs = tokenizer(input_text, return_tensors="pt")
9outputs = model.generate(**inputs, max_length=1000)
10response = tokenizer.decode(outputs[0])
11print(response)@misc{OncoCareBrain-GPT,
author = {DXCLab},
title = {OncoCareBrain-GPT: A Specialized Language Model for Oncology},
year = {2025},
publisher = {Hugging Face},
howpublished = {\url{https://huggingface.co/DXCLab/OncoCareBrain-GPT}}
}