Create README.md

README.md

@@ -0,0 +1,35 @@
+# 🧠 Disease Symptom Classifier
+
+A BERT-based model for classifying text descriptions of symptoms into medical-related categories.
+
+## 🔧 Usage
+
+```python
+from transformers import pipeline
+
+pipe = pipeline("text-classification", model="Luwayy/disease_predict4")
+
+text = "The patient is experiencing a persistent cough and shortness of breath."
+result = pipe(text)
+print(result)  # [{'label': 'Cough', 'score': 0.9985}]
+```
+
+## 🏷️ Labels
+
+The model predicts one of the following symptoms:
+
+```
+- Emotional pain           - Joint pain
+- Hair falling out         - Hard to breath
+- Heart hurts              - Head ache
+- Infected wound           - Body feels weak
+- Foot ache                - Feeling dizzy
+- Shoulder pain            - Back pain
+- Injury from sports       - Open wound
+- Skin issue               - Internal pain
+- Stomach ache             - Blurry vision
+- Knee pain                - Acne
+- Muscle pain              - Neck pain
+- Cough                    - Ear ache
+- Feeling cold
+```