# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Hayato-T08/disease_diagnosis")
model = AutoModelForSequenceClassification.from_pretrained("Hayato-T08/disease_diagnosis", device_map="auto")LABEL2disease
('0', '(vertigo) Paroymsal Positional Vertigo')
('1', 'AIDS')
('2', 'Acne')
('3', 'Alcoholic hepatitis')
('4', 'Allergy')
('5', 'Arthritis')
('6', 'Bronchial Asthma')
('7', 'Cervical spondylosis')
('8', 'Chicken pox')
('9', 'Chronic cholestasis')
('10', 'Common Cold')
('11', 'Dengue')
('12', 'Diabetes ')
('13', 'Dimorphic hemmorhoids(piles)')
('14', 'Drug Reaction')
('15', 'Fungal infection')
('16', 'GERD')
('17', 'Gastroenteritis')
('18', 'Heart attack')
('19', 'Hepatitis B')
('20', 'Hepatitis C')
('21', 'Hepatitis D')
('22', 'Hepatitis E')
('23', 'Hypertension ')
('24', 'Hyperthyroidism')
('25', 'Hypoglycemia')
('26', 'Hypothyroidism')
('27', 'Impetigo')
('28', 'Jaundice')
('29', 'Malaria')
('30', 'Migraine')
('31', 'Osteoarthristis')
('32', 'Paralysis (brain hemorrhage)')
('33', 'Peptic ulcer diseae')
('34', 'Pneumonia')
('35', 'Psoriasis')
('36', 'Tuberculosis')
('37', 'Typhoid')
('38', 'Urinary tract infection')
('39', 'Varicose veins')
('40', 'hepatitis A')
- Downloads last month
- 5
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Hayato-T08/disease_diagnosis")