konradpieszko commited on
Commit
c5fc1bd
·
1 Parent(s): c8220b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ from my_waterfall import my_waterfall
15
 
16
  def make_prediction(age, arrythmia_duration_radio,TEE_base_rythm_radio,HF_status_radio, NYHA_radio,EF, LA_dimension,LAVI,):
17
  plt.clf()
18
- names=["Age", "Arrhythmia type", "Rhytm", "Heart failure", "NYHA class", "LVEF", "LA dimension", "LAVI"]
19
  with open("LATAI_reduced_model.pickle", "rb") as f:
20
  clf = pickle.load(f)
21
  #print(arrythmia_duration_radio)
 
15
 
16
  def make_prediction(age, arrythmia_duration_radio,TEE_base_rythm_radio,HF_status_radio, NYHA_radio,EF, LA_dimension,LAVI,):
17
  plt.clf()
18
+ names=['Age', 'arrythmia_duration', 'LA_dimension', 'LAVI', 'NYHA', 'HF_status', 'TEE_base_rythm', 'EF']
19
  with open("LATAI_reduced_model.pickle", "rb") as f:
20
  clf = pickle.load(f)
21
  #print(arrythmia_duration_radio)