Mohit8219 commited on
Commit
e2a64f7
·
verified ·
1 Parent(s): 48e6b74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -9,6 +9,9 @@ import gradio as gr
9
  # Load dataset
10
  df = pd.read_csv("hf://datasets/buio/heart-disease/heart.csv")
11
 
 
 
 
12
  # Define features and target
13
  X = df.drop('target', axis=1)
14
  y = df['target']
 
9
  # Load dataset
10
  df = pd.read_csv("hf://datasets/buio/heart-disease/heart.csv")
11
 
12
+ # Convert categorical columns to numeric using one-hot encoding
13
+ df = pd.get_dummies(df, drop_first=True)
14
+
15
  # Define features and target
16
  X = df.drop('target', axis=1)
17
  y = df['target']