Question Answering
Transformers
Safetensors
PyTorch
gpt2
text-generation
dialogpt
conversational
medical
text-generation-inference
Instructions to use anique-1/finetuned-dialogpt-medium with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use anique-1/finetuned-dialogpt-medium with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("question-answering", model="anique-1/finetuned-dialogpt-medium") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("anique-1/finetuned-dialogpt-medium") model = AutoModelForCausalLM.from_pretrained("anique-1/finetuned-dialogpt-medium") messages = [ {"role": "user", "content": "Who are you?"}, ] inputs = tokenizer.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=40) print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Notebooks
- Google Colab
- Kaggle
Finetuned DialoGPT-medium for Medical QA
This is a version of Microsoft's DialoGPT-medium model finetuned for medical question answering using a collection of medical QA datasets.
Model Details
- Base Model: microsoft/DialoGPT-medium
- Finetuned for: Medical Question Answering (QA)
- Datasets:
- CancerQA
- Diabetes and Digestive and Kidney Diseases QA
- Genetic and Rare Diseases QA
- Growth Hormone Receptor QA
- Heart, Lung, and Blood QA
- Medical Question Answering
- Neurological Disorders and Stroke QA
- Other QA
- Senior Health QA
Usage
You can use this model with the Hugging Face transformers library:
from transformers import pipeline
pipe = pipeline("text-generation", model="anique-1/finetuned-dialogpt-medium")
messages = [
{"role": "user", "content": "What are the symptoms of diabetes?"},
]
response = pipe(messages)
print(response)
Or, if using locally:
pipe = pipeline("text-generation", model="path/to/finetuned_DialoGPT_medium_safetensors")
Intended Use
- Primary: Medical question answering for research and educational purposes.
- Not for: Clinical decision making or as a substitute for professional medical advice.
Limitations
- The model may generate incorrect or misleading medical information.
- Not a replacement for professional healthcare advice.
License
Apache 2.0
Citation
If you use this model, please cite the base model and this repository.
Contact
For questions or feedback, please contact the model maintainer.
- Downloads last month
- 2