apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

do cors correctly

This commit is contained in:
Sean Sube 2023-01-04 22:54:17 -06:00
parent 513d0a4d0f
commit c4e542bc40
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/serve.py
View File

@ -1,5 +1,5 @@
from diffusers import OnnxStableDiffusionPipeline from diffusers import OnnxStableDiffusionPipeline
from flask import Flask, request, response, send_file from flask import Flask, make_response, request, send_file
from io import BytesIO from io import BytesIO
from os import environ, path, makedirs from os import environ, path, makedirs
@ -57,5 +57,6 @@ def txt2img():
image_queue.remove(user) image_queue.remove(user)
response.headers.add('Access-Control-Allow-Origin', '*') res = make_response(send_file(img_io, mimetype='image/png'))
return send_file(img_io, mimetype='image/png') res.headers.add('Access-Control-Allow-Origin', '*')
return res