-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.py
25 lines (18 loc) · 617 Bytes
/
server.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
import joblib
import numpy as np
from flask import Flask, jsonify
app = Flask(__name__)
import joblib
import numpy as np
from flask import Flask
from flask import jsonify
app = Flask(__name__)
#POSTMAN PARA PRUEBAS
@app.route('/predict', methods=['GET'])
def predict():
X_test = np.array([7.594444821,7.479555538,1.616463184,1.53352356,0.796666503,0.635422587,0.362012237,0.315963835,2.277026653])
prediction = model.predict(X_test.reshape(1,-1))
return jsonify({'prediccion' : list(prediction)})
if __name__ == "__main__":
model = joblib.load('./models/best_model.pkl')
app.run(port=8080)