fmegahed commited on
Commit
cf9b184
·
1 Parent(s): cee1164

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -13,8 +13,12 @@ import numpy as np
13
  from io import BytesIO
14
  import pickle
15
  import requests
 
16
 
17
- ex_data = pd.read_csv('https://raw.githubusercontent.com/fmegahed/tavr_paper/main/data/example_data2.csv')
 
 
 
18
  ex_data = ex_data.to_numpy()
19
  ex_data = ex_data.tolist()
20
 
@@ -30,10 +34,9 @@ def predict(age, female, race, elective, aweekend, zipinc_qrtl, hosp_region, hos
30
  pulmonary_circulation_disorder, smoker, valvular_disease, weight_loss,
31
  endovascular_tavr, transapical_tavr):
32
 
33
- mLink = 'https://github.com/fmegahed/tavr_paper/blob/main/data/final_model.pkl?raw=true'
34
- mfile = BytesIO(requests.get(mLink).content)
35
- model = pickle.load(mfile)
36
- model = load_model(model)
37
 
38
  df = pd.DataFrame.from_dict({
39
  'age': [age], 'female': [female], 'race': [race], 'elective': elective,
 
13
  from io import BytesIO
14
  import pickle
15
  import requests
16
+ from urllib.request import urlopen
17
 
18
+ url = 'https://raw.githubusercontent.com/fmegahed/tavr_paper/main/data/example_data2.csv'
19
+ download = requests.get(url).content
20
+
21
+ ex_data =pd.read_csv(io.StringIO(download.decode('utf-8')))
22
  ex_data = ex_data.to_numpy()
23
  ex_data = ex_data.tolist()
24
 
 
34
  pulmonary_circulation_disorder, smoker, valvular_disease, weight_loss,
35
  endovascular_tavr, transapical_tavr):
36
 
37
+ url = 'https://github.com/fmegahed/tavr_paper/blob/main/data/final_model.pkl?raw=true'
38
+ raw = urlopen(url)
39
+ model = pickle.load(raw)
 
40
 
41
  df = pd.DataFrame.from_dict({
42
  'age': [age], 'female': [female], 'race': [race], 'elective': elective,