sid570 commited on
Commit
ffe2ca2
·
verified ·
1 Parent(s): 2404fb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -2,11 +2,12 @@ import gradio as gr
2
  import joblib
3
  import numpy as np
4
 
5
- # Load model
6
- model = joblib.load("RF_model_50_diseases.pkl")
7
 
8
- # Load symptom columns
9
- symptoms = joblib.load("symptom_columns.pkl")
 
 
10
 
11
  def predict_disease(selected_symptoms):
12
 
 
2
  import joblib
3
  import numpy as np
4
 
5
+ BUNDLE_PATH = "RF_bundle_50_diseases.pkl"
 
6
 
7
+ bundle = joblib.load(BUNDLE_PATH)
8
+
9
+ model = bundle["model"]
10
+ symptoms = bundle["features"]
11
 
12
  def predict_disease(selected_symptoms):
13