From 45cc2784cd4239cc8171060599b473e204e9a2ed Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Wed, 4 Jan 2023 23:44:16 -0600 Subject: [PATCH] add extension to output files --- api/serve.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/serve.py b/api/serve.py index a977507a..6bd4fe6a 100644 --- a/api/serve.py +++ b/api/serve.py @@ -9,7 +9,7 @@ from diffusers import ( DPMSolverMultistepScheduler, ) from flask import Flask, make_response, request, send_file -from stringcase import snakecase +from stringcase import spinalcase from io import BytesIO from os import environ, path, makedirs @@ -83,7 +83,7 @@ def txt2img(): guidance_scale=cfg ).images[0] - output = '%s/txt2img-%s' % (output_path, snakecase(prompt)) + output = '%s/txt2img_%s.png' % (output_path, spinalcase(prompt)) print("txt2img output: %s" % (output)) image.save(output)