zhaojinghao
/
ai_platform_regression
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

修改response

This commit is contained in:
zhaojinghao 2022-12-09 18:37:19 +08:00
parent b387ce871d
commit 4f101eec52
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
run.py
View File

@ -6,13 +6,13 @@ import json
from logzero import logger from logzero import logger
import io import io
from house_price.house_price_predcition import run_boston_price # from house_price.house_price_predcition import run_boston_price
from ocean_wave.wave_height_mlp import predict_wave_height # from ocean_wave.wave_height_mlp import predict_wave_height
from prophet_predict.prophet_predict import run_prophet from prophet_predict.prophet_predict import run_prophet
TEXT = "text" TEXT = "text"
app = Flask(__name__) app = Flask(__name__)
'''
@app.route('/house_price/', methods=["POST"]) @app.route('/house_price/', methods=["POST"])
def predict_price(): def predict_price():
@ -62,7 +62,7 @@ def predict_price():
if resp_info["code"] == 200: if resp_info["code"] == 200:
resp = make_response(resp_info["data"]) resp = make_response(resp_info["data"])
resp.headers["Content-Disposition"] = "attachment; filename=house_price.xlsx" resp.headers["Content-Disposition"] = "attachment; filename=house_price.xlsx"
resp.headers["Content-type"] = "application/x-xlsx" resp.headers["Content-type"] = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8"
else: else:
resp = make_response(json.dumps(resp_info)) resp = make_response(json.dumps(resp_info))
resp.status_code = 200 resp.status_code = 200
@ -112,7 +112,7 @@ def predict_height():
resp = make_response(json.dumps(resp_info)) resp = make_response(json.dumps(resp_info))
resp.status_code = 200 resp.status_code = 200
return resp return resp
'''
@app.route("/prophet/", methods=["POST"]) @app.route("/prophet/", methods=["POST"])
def run_ts_predict(): def run_ts_predict():
@ -147,7 +147,7 @@ def run_ts_predict():
if resp_info.get("code") == 200: if resp_info.get("code") == 200:
resp = make_response(resp_info["data"]) resp = make_response(resp_info["data"])
resp.headers["Content-Disposition"] = f"attachment; filename={file_name}" resp.headers["Content-Disposition"] = f"attachment; filename={file_name}"
resp.headers["Content-type"] = "application/x-xlsx" resp.headers["Content-type"] = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8"
else: else:
resp = make_response(json.dumps(resp_info)) resp = make_response(json.dumps(resp_info))
resp.status_code = 200 resp.status_code = 200