alibidaran commited on
Commit
37b924f
1 Parent(s): b261e83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,6 +1,11 @@
1
  import pickle
2
  import gradio as gr
3
  import numpy as np
 
 
 
 
 
4
  file_name='heart_disease_model.sav'
5
  model=pickle.load(open('heart_disease_model.sav','rb'))
6
  def show_output(age,sex,chestpain,maxhr,st_slope,cholestrol):
 
1
  import pickle
2
  import gradio as gr
3
  import numpy as np
4
+ import pandas as pd
5
+ df=pd.read_csv("heart.csv")
6
+ encode_sex={name:i for i,name in enumerate(df["Sex"].unique())}
7
+ encode_chest={name:i for i,name in enumerate(df["ChestPainType"].unique())}
8
+ encode_slope={name:i for i, name in enumerate(df['ST_Slope'].unique())}
9
  file_name='heart_disease_model.sav'
10
  model=pickle.load(open('heart_disease_model.sav','rb'))
11
  def show_output(age,sex,chestpain,maxhr,st_slope,cholestrol):