fmegahed commited on
Commit
e732af4
·
1 Parent(s): 7d02cf0

Update app.py to include data from GitHub

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,6 +1,7 @@
1
  # pip install pycaret
2
  import pandas as pd
3
  import jinja2
 
4
 
5
  from pycaret.classification import *
6
  import imblearn as im
@@ -9,7 +10,7 @@ import sklearn
9
  import gradio as gr
10
  import numpy as np
11
 
12
- ex_data = pd.read_csv('example_data.csv')
13
  ex_data = ex_data.to_numpy()
14
  ex_data = ex_data.tolist()
15
 
@@ -25,7 +26,7 @@ def predict(age, female, race, elective, aweekend, zipinc_qrtl, hosp_region, hos
25
  pulmonary_circulation_disorder, smoker, valvular_disease, weight_loss,
26
  endovascular_tavr, transapical_tavr):
27
 
28
- model = load_model('final_model')
29
 
30
  df = pd.DataFrame.from_dict({
31
  'age': [age], 'female': [female], 'race': [race], 'elective': elective,
 
1
  # pip install pycaret
2
  import pandas as pd
3
  import jinja2
4
+ from datasets import load_dataset
5
 
6
  from pycaret.classification import *
7
  import imblearn as im
 
10
  import gradio as gr
11
  import numpy as np
12
 
13
+ ex_data = pd.read_csv('https://raw.githubusercontent.com/fmegahed/tavr_paper/main/data/example_data2.csv')
14
  ex_data = ex_data.to_numpy()
15
  ex_data = ex_data.tolist()
16
 
 
26
  pulmonary_circulation_disorder, smoker, valvular_disease, weight_loss,
27
  endovascular_tavr, transapical_tavr):
28
 
29
+ model = load_model('https://github.com/fmegahed/tavr_paper/blob/main/data/final_model.pkl?raw=true')
30
 
31
  df = pd.DataFrame.from_dict({
32
  'age': [age], 'female': [female], 'race': [race], 'elective': elective,