MedLLM / README.md
shivvamm's picture
Update README.md
a115c8b verified
|
raw
history blame
1.43 kB
metadata
license: mit
datasets:
  - QuyenAnhDE/Diseases_Symptoms
language:
  - en
pipeline_tag: text-generation
library_name: transformers

MedLLM

Model Description

MedLLM is a language model designed to assist healthcare professionals and patients by providing detailed information about the symptoms of various diseases. This model can help in preliminary assessments and educational purposes by offering accurate and concise symptom descriptions.

Model Details

  • Architecture: Transformer-based architecture optimized for medical text.
  • Training Data: Trained on a diverse set of medical texts, including clinical notes, research articles, and symptom databases, to ensure a comprehensive understanding of disease symptoms.
  • Training Procedure: Fine-tuned using supervised learning techniques with a focus on accuracy and relevancy in medical contexts. Hyperparameters were adjusted for optimal performance in natural language understanding and generation.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "shivvamm/MedLLM"  # Change to your repo name
model = AutoModelForCausalLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

inputs = tokenizer("What are the symptoms of diabetes?", return_tensors="pt")
outputs = model.generate(**inputs)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(response)