cogniveon commited on
Commit
d24a308
1 Parent(s): a8cee8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -49,20 +49,20 @@ def get_patient_outcome(age, sex, height, weight, is_pregnant, murmur):
49
  'Murmur': float(murmur),
50
  }, index=[0])
51
 
52
- # Predict Keras
53
- output = keras_model.predict(data)[0]
54
- # 0 - Normal, 1 - Abnormal -> %
55
- results_keras = {
56
- 'Normal': output[0],
57
- 'Abnormal': output[1]
58
- }
59
 
60
  # Predict SVC
61
  output = model.predict(data)[0]
62
  # 0 - Normal or 1 - Abnormal
63
  results_svc = 'Normal' if output == 1 else 'Abnormal'
64
 
65
- return results_keras, results_svc
66
 
67
 
68
  def predict(audio, age, sex, height, weight, is_pregnant):
@@ -94,7 +94,6 @@ demo = gr.Interface(
94
  gr.Checkbox(label="Pregnant", value=False)
95
  ],
96
  outputs=[
97
- gr.Label(label="keras_pred"),
98
  gr.Label(label="svc_pred"),
99
  ],
100
  cache_examples=True,
 
49
  'Murmur': float(murmur),
50
  }, index=[0])
51
 
52
+ # # Predict Keras
53
+ # output = keras_model.predict(data)[0]
54
+ # # 0 - Normal, 1 - Abnormal -> %
55
+ # results_keras = {
56
+ # 'Normal': output[0],
57
+ # 'Abnormal': output[1]
58
+ # }
59
 
60
  # Predict SVC
61
  output = model.predict(data)[0]
62
  # 0 - Normal or 1 - Abnormal
63
  results_svc = 'Normal' if output == 1 else 'Abnormal'
64
 
65
+ return results_svc
66
 
67
 
68
  def predict(audio, age, sex, height, weight, is_pregnant):
 
94
  gr.Checkbox(label="Pregnant", value=False)
95
  ],
96
  outputs=[
 
97
  gr.Label(label="svc_pred"),
98
  ],
99
  cache_examples=True,