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
4
Safetensors
Model size
0.4B params
Tensor type
F32
Inference Providers NEW
This model isn't deployed by any Inference Provider. 馃檵 Ask for provider support

Evaluation results