feat(api): add endpoint to get previous outputs
This commit is contained in:
parent
668e46a404
commit
50221af55a
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue