add extension to output files
This commit is contained in:
parent
2e95e02ef1
commit
45cc2784cd
|
@ -9,7 +9,7 @@ from diffusers import (
|
||||||
DPMSolverMultistepScheduler,
|
DPMSolverMultistepScheduler,
|
||||||
)
|
)
|
||||||
from flask import Flask, make_response, request, send_file
|
from flask import Flask, make_response, request, send_file
|
||||||
from stringcase import snakecase
|
from stringcase import spinalcase
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
from os import environ, path, makedirs
|
from os import environ, path, makedirs
|
||||||
|
|
||||||
|
@ -83,7 +83,7 @@ def txt2img():
|
||||||
guidance_scale=cfg
|
guidance_scale=cfg
|
||||||
).images[0]
|
).images[0]
|
||||||
|
|
||||||
output = '%s/txt2img-%s' % (output_path, snakecase(prompt))
|
output = '%s/txt2img_%s.png' % (output_path, spinalcase(prompt))
|
||||||
print("txt2img output: %s" % (output))
|
print("txt2img output: %s" % (output))
|
||||||
image.save(output)
|
image.save(output)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue