-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathflaskServer.py
41 lines (34 loc) · 1.1 KB
/
flaskServer.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
from flask import Flask, request, jsonify
from flask import Flask, request, jsonify
from Ml import make_prediction
app = Flask(__name__)
@app.route('/fapi/predict', methods=['POST'])
def predict():
try:
# Get the patient data from the request
data = request.get_json()
# df = pd.DataFrame(data)
print("********************************")
# print(data)
new_data = {
'Age': [data["age"]],
'Gender': [data["gender"]],
'Heart rate': [data["heartRate"]],
'Systolic blood pressure': [data["systolicBP"]],
'Diastolic blood pressure': [data["diastolicBP"]],
'Blood sugar': [data["bloodSugar"]],
'CK-MB': [data["ckMb"]],
'Troponin': [data["troponin"]],
}
result = make_prediction(new_data)
print("Prediction:", result)
print("********************************")
prediction = {
'message': result[0]
}
# Return the dummy prediction
return jsonify(prediction)
except Exception as e:
return jsonify({'error': str(e)}), 500
if __name__ == '__main__':
app.run(debug=True)