Duplicate from Dikshant09/disease-prediction-api

Co-authored-by: Dikshant <Dikshant09@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: Disease Prediction Api
+emoji: 📊
+colorFrom: indigo
+colorTo: green
+sdk: gradio
+sdk_version: 3.19.1
+app_file: app.py
+pinned: false
+license: mit
+duplicated_from: Dikshant09/disease-prediction-api
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,56 @@
+# Importing libraries
+import numpy as np
+from scipy.stats import mode
+import warnings
+import pickle
+
+import gradio as gr
+
+# Ignoring Warnings
+warnings.filterwarnings("ignore")
+
+# Loading Saved Models
+model = pickle.load(open('saved_final_rf_model.pkl', 'rb'))
+
+data_dict = {'symptom_index': {'Itching': 0, 'Skin Rash': 1, 'Nodal Skin Eruptions': 2, 'Continuous Sneezing': 3, 'Shivering': 4, 'Chills': 5, 'Joint Pain': 6, 'Stomach Pain': 7, 'Acidity': 8, 'Ulcers On Tongue': 9, 'Muscle Wasting': 10, 'Vomiting': 11, 'Burning Micturition': 12, 'Spotting  urination': 13, 'Fatigue': 14, 'Weight Gain': 15, 'Anxiety': 16, 'Cold Hands And Feets': 17, 'Mood Swings': 18, 'Weight Loss': 19, 'Restlessness': 20, 'Lethargy': 21, 'Patches In Throat': 22, 'Irregular Sugar Level': 23, 'Cough': 24, 'High Fever': 25, 'Sunken Eyes': 26, 'Breathlessness': 27, 'Sweating': 28, 'Dehydration': 29, 'Indigestion': 30, 'Headache': 31, 'Yellowish Skin': 32, 'Dark Urine': 33, 'Nausea': 34, 'Loss Of Appetite': 35, 'Pain Behind The Eyes': 36, 'Back Pain': 37, 'Constipation': 38, 'Abdominal Pain': 39, 'Diarrhoea': 40, 'Mild Fever': 41, 'Yellow Urine': 42, 'Yellowing Of Eyes': 43, 'Acute Liver Failure': 44, 'Fluid Overload': 45, 'Swelling Of Stomach': 46, 'Swelled Lymph Nodes': 47, 'Malaise': 48, 'Blurred And Distorted Vision': 49, 'Phlegm': 50, 'Throat Irritation': 51, 'Redness Of Eyes': 52, 'Sinus Pressure': 53, 'Runny Nose': 54, 'Congestion': 55, 'Chest Pain': 56, 'Weakness In Limbs': 57, 'Fast Heart Rate': 58, 'Pain During Bowel Movements': 59, 'Pain In Anal Region': 60, 'Bloody Stool': 61, 'Irritation In Anus': 62, 'Neck Pain': 63, 'Dizziness': 64, 'Cramps': 65, 'Bruising': 66, 'Obesity': 67, 'Swollen Legs': 68, 'Swollen Blood Vessels': 69, 'Puffy Face And Eyes': 70, 'Enlarged Thyroid': 71, 'Brittle Nails': 72, 'Swollen Extremeties': 73, 'Excessive Hunger': 74, 'Extra Marital Contacts': 75, 'Drying And Tingling Lips': 76, 'Slurred Speech': 77, 'Knee Pain': 78, 'Hip Joint Pain': 79, 'Muscle Weakness': 80, 'Stiff Neck': 81, 'Swelling Joints': 82, 'Movement Stiffness': 83, 'Spinning Movements': 84, 'Loss Of Balance': 85, 'Unsteadiness': 86, 'Weakness Of One Body Side': 87, 'Loss Of Smell': 88, 'Bladder Discomfort': 89, 'Foul Smell Of urine': 90, 'Continuous Feel Of Urine': 91, 'Passage Of Gases': 92, 'Internal Itching': 93, 'Toxic Look (typhos)': 94, 'Depression': 95, 'Irritability': 96, 'Muscle Pain': 97, 'Altered Sensorium': 98, 'Red Spots Over Body': 99, 'Belly Pain': 100, 'Abnormal Menstruation': 101, 'Dischromic  Patches': 102, 'Watering From Eyes': 103, 'Increased Appetite': 104, 'Polyuria': 105, 'Family History': 106, 'Mucoid Sputum': 107, 'Rusty Sputum': 108, 'Lack Of Concentration': 109, 'Visual Disturbances': 110, 'Receiving Blood Transfusion': 111, 'Receiving Unsterile Injections': 112, 'Coma': 113, 'Stomach Bleeding': 114, 'Distention Of Abdomen': 115, 'History Of Alcohol Consumption': 116, 'Fluid Overload.1': 117, 'Blood In Sputum': 118, 'Prominent Veins On Calf': 119, 'Palpitations': 120, 'Painful Walking': 121, 'Pus Filled Pimples': 122, 'Blackheads': 123, 'Scurring': 124, 'Skin Peeling': 125, 'Silver Like Dusting': 126, 'Small Dents In Nails': 127, 'Inflammatory Nails': 128, 'Blister': 129, 'Red Sore Around Nose': 130, 'Yellow Crust Ooze': 131}, 
+	     'predictions_classes': ['(vertigo) Paroymsal  Positional Vertigo', 'AIDS', 'Acne',
+       'Alcoholic hepatitis', 'Allergy', 'Arthritis', 'Bronchial Asthma',
+       'Cervical spondylosis', 'Chicken pox', 'Chronic cholestasis',
+       'Common Cold', 'Dengue', 'Diabetes ',
+       'Dimorphic hemmorhoids(piles)', 'Drug Reaction',
+       'Fungal infection', 'GERD', 'Gastroenteritis', 'Heart attack',
+       'Hepatitis B', 'Hepatitis C', 'Hepatitis D', 'Hepatitis E',
+       'Hypertension ', 'Hyperthyroidism', 'Hypoglycemia',
+       'Hypothyroidism', 'Impetigo', 'Jaundice', 'Malaria', 'Migraine',
+       'Osteoarthristis', 'Paralysis (brain hemorrhage)',
+       'Peptic ulcer diseae', 'Pneumonia', 'Psoriasis', 'Tuberculosis',
+       'Typhoid', 'Urinary tract infection', 'Varicose veins',
+       'hepatitis A']}
+
+def predictDisease(symptoms):
+	symptoms = symptoms.split(",")
+
+	# creating input data for the models
+	input_data = [0] * len(data_dict["symptom_index"])
+	for symptom in symptoms:
+		index = data_dict["symptom_index"][symptom]
+		input_data[index] = 1
+		
+	# reshaping the input data and converting it
+	# into suitable format for model predictions
+	input_data = np.array(input_data).reshape(1,-1)
+	
+	# generating individual outputs
+	prediction = data_dict["predictions_classes"][model.predict(input_data)[0]]
+	
+	# making final prediction by taking mode of all predictions
+	final_prediction = mode([prediction])[0][0]
+	return final_prediction
+
+# Input: Systoms
+# Output: Disease
+def makePrediction(Enter_Symptom):
+    return predictDisease(Enter_Symptom)
+
+iface = gr.Interface(fn = makePrediction, inputs = "text", outputs = "text")
+iface.launch()

requirements.txt

@@ -0,0 +1,32 @@
+certifi==2022.12.7
+charset-normalizer==3.0.1
+contourpy==1.0.7
+cycler==0.11.0
+docopt==0.6.2
+filelock==3.9.0
+fonttools==4.38.0
+huggingface-hub==0.12.1
+idna==3.4
+joblib==1.2.0
+kiwisolver==1.4.4
+matplotlib==3.6.3
+numpy==1.24.2
+packaging==23.0
+pandas==1.5.3
+Pillow==9.4.0
+pipreqs==0.4.11
+pyparsing==3.0.9
+python-dateutil==2.8.2
+pytz==2022.7.1
+PyYAML==6.0
+requests==2.28.2
+scikit-learn==1.2.1
+scipy==1.10.0
+seaborn==0.12.2
+six==1.16.0
+sklearn==0.0.post1
+threadpoolctl==3.1.0
+tqdm==4.64.1
+typing_extensions==4.5.0
+urllib3==1.26.14
+yarg==0.1.9

saved_final_nb_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:322440f78fbc11977feac3624afbe0f767f63690bfd86740e70465eb397b53fd
+size 90359

saved_final_rf_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ff7ce105305bd240bf93f7db7916dbd08bbc7dd9f6fc0564b3eeb1fba072ea3
+size 7002340

saved_final_svm_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ab332c2a0bb83fa1261644e60af7075168096183810cbab3d34e070973964d5
+size 454711