CoolFace
Modelpublic

Luwayy/disease_predict4

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes11downloads
Model Card

🧠 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