apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): add endpoint to get previous outputs

This commit is contained in:
Sean Sube 2023-01-05 17:24:33 -06:00
parent 668e46a404
commit 50221af55a
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/serve.py
View File

@ -8,7 +8,7 @@ from diffusers import (
LMSDiscreteScheduler, LMSDiscreteScheduler,
PNDMScheduler, PNDMScheduler,
) )
from flask import Flask, make_response, request, send_file from flask import Flask, make_response, request, send_file, send_from_directory
from stringcase import spinalcase from stringcase import spinalcase
from io import BytesIO from io import BytesIO
from os import environ, path, makedirs from os import environ, path, makedirs
@ -135,3 +135,7 @@ def txt2img():
res = make_response(send_file(img_io, mimetype='image/png')) res = make_response(send_file(img_io, mimetype='image/png'))
res.headers.add('Access-Control-Allow-Origin', '*') res.headers.add('Access-Control-Allow-Origin', '*')
return res return res
@app.route('/output/<path:filename>')
def output(filename):
return send_from_directory(output_path, filename, as_attachment=False)