cogniveon commited on
Commit
fb6222e
·
verified ·
1 Parent(s): e646c75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -53,8 +53,8 @@ def get_patient_outcome(age, sex, height, weight, is_pregnant, murmur):
53
  output = model.predict(data)[0]
54
  # 0 - Normal, 1 - Abnormal
55
  results = {
56
- 'Normal': output[0],
57
- 'Abnormal': output[1]
58
  }
59
 
60
  return results
 
53
  output = model.predict(data)[0]
54
  # 0 - Normal, 1 - Abnormal
55
  results = {
56
+ 'Normal': output == 0,
57
+ 'Abnormal': output == 1
58
  }
59
 
60
  return results