Shivam commited on
Commit
4370f74
1 Parent(s): 0dc37d6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +33 -3
README.md CHANGED
@@ -1,3 +1,33 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ datasets:
4
+ - QuyenAnhDE/Diseases_Symptoms
5
+ language:
6
+ - en
7
+ pipeline_tag: text-generation
8
+ library_name: transformers
9
+ ---
10
+
11
+ # MedLLM
12
+
13
+ ## Model Description
14
+ 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.
15
+
16
+ ## Model Details
17
+ - **Architecture**: Transformer-based architecture optimized for medical text.
18
+ - **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.
19
+ - **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.
20
+
21
+ ## Usage
22
+ ```python
23
+ from transformers import AutoModelForCausalLM, AutoTokenizer
24
+
25
+ model_name = "your-username/MedLLaMA" # Change to your repo name
26
+ model = AutoModelForCausalLM.from_pretrained(model_name)
27
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
28
+
29
+ inputs = tokenizer("What are the symptoms of diabetes?", return_tensors="pt")
30
+ outputs = model.generate(**inputs)
31
+ response = tokenizer.decode(outputs[0], skip_special_tokens=True)
32
+ print(response)
33
+ ```