Pranjal-psytech commited on
Commit
896a005
1 Parent(s): 36b17f5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ loaded_model = joblib.load('heart.pkl')
18
  def heart_disease(age, sex, cp, trestbps, chol, fbs, restecg, thalach, exang, oldpeak, slope, ca,thal):
19
  #turning the arguments into a numpy array
20
 
21
- x = np.array([age, sex, cp, trestbps, chol, fbs, restecg, thalach, exang, oldpeak, slope, ca,thal])
22
 
23
  prediction = loaded_model.predict(x.reshape(1, -1))
24
 
 
18
  def heart_disease(age, sex, cp, trestbps, chol, fbs, restecg, thalach, exang, oldpeak, slope, ca,thal):
19
  #turning the arguments into a numpy array
20
 
21
+ x = np.array([age, sex, cp, trestbps, chol, fbs, restecg, thalach, exang, oldpeak, slope, ca,thal],dtype=float)
22
 
23
  prediction = loaded_model.predict(x.reshape(1, -1))
24