Spaces:

fmegahed
/

tavr_project

Runtime error

fmegahed commited on May 27, 2022

Commit

6bf09c1

1 Parent(s): 21106dc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,8 @@ def predict(age, female, race, elective, aweekend, zipinc_qrtl, hosp_region, hos
   mLink = 'https://github.com/fmegahed/tavr_paper/blob/main/data/final_model.pkl?raw=true'
   mfile = BytesIO(requests.get(mLink).content)
-  model = load_model(mfile)
   df = pd.DataFrame.from_dict({
       'age': [age], 'female': [female], 'race': [race], 'elective': elective,

   mLink = 'https://github.com/fmegahed/tavr_paper/blob/main/data/final_model.pkl?raw=true'
   mfile = BytesIO(requests.get(mLink).content)
+  model = pickle.load(mfile)
+  model = load_model(model)
   df = pd.DataFrame.from_dict({
       'age': [age], 'female': [female], 'race': [race], 'elective': elective,