From c4e542bc40097943f07219555cc7b04e50d4a812 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Wed, 4 Jan 2023 22:54:17 -0600 Subject: [PATCH] do cors correctly --- api/serve.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/api/serve.py b/api/serve.py index 4ad1cf74..b49ddcb9 100644 --- a/api/serve.py +++ b/api/serve.py @@ -1,5 +1,5 @@ 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 os import environ, path, makedirs @@ -57,5 +57,6 @@ def txt2img(): image_queue.remove(user) - response.headers.add('Access-Control-Allow-Origin', '*') - return send_file(img_io, mimetype='image/png') \ No newline at end of file + res = make_response(send_file(img_io, mimetype='image/png')) + res.headers.add('Access-Control-Allow-Origin', '*') + return res \ No newline at end of file