| # 🧠 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 | |
| ``` | |