frogleo commited on
Commit
7cd4941
·
1 Parent(s): 063d6c4

强制改为jpg格式

Browse files
__pycache__/config.cpython-310.pyc DELETED
Binary file (1.04 kB)
 
__pycache__/utils.cpython-310.pyc DELETED
Binary file (3.78 kB)
 
app.py CHANGED
@@ -177,12 +177,10 @@ def generate(
177
  callback_on_step_end=callback2
178
  ).images
179
  out_img = images[0]
180
- out_img = out_img.convert("RGB")
181
- img_io = io.BytesIO()
182
- out_img.save(img_io, format="PNG") # 使用 PNG 格式保存
183
- img_io.seek(0)
184
  progress(1, desc="Complete")
185
- return Image.open(img_io)
186
  except GenerationError as e:
187
  logger.warning(f"Generation validation error: {str(e)}")
188
  raise gr.Error(str(e))
@@ -309,6 +307,7 @@ with gr.Blocks(css=custom_css).queue() as demo:
309
  with gr.Column():
310
  gr.Markdown("### Output")
311
  result = gr.Image(
 
312
  label="Generated Image",
313
  elem_id="output-image"
314
  )
 
177
  callback_on_step_end=callback2
178
  ).images
179
  out_img = images[0]
180
+ path = utils.save_image(out_img, "./outputs")
181
+ logger.info(f"output path: {path}")
 
 
182
  progress(1, desc="Complete")
183
+ return path
184
  except GenerationError as e:
185
  logger.warning(f"Generation validation error: {str(e)}")
186
  raise gr.Error(str(e))
 
307
  with gr.Column():
308
  gr.Markdown("### Output")
309
  result = gr.Image(
310
+ type="filepath",
311
  label="Generated Image",
312
  elem_id="output-image"
313
  )
utils.py CHANGED
@@ -99,4 +99,11 @@ def free_memory() -> None:
99
  if torch.cuda.is_available():
100
  torch.cuda.empty_cache()
101
  torch.cuda.ipc_collect()
102
- gc.collect()
 
 
 
 
 
 
 
 
99
  if torch.cuda.is_available():
100
  torch.cuda.empty_cache()
101
  torch.cuda.ipc_collect()
102
+ gc.collect()
103
+
104
+ def save_image(image, output_dir):
105
+ filename = str(uuid.uuid4()) + ".jpg"
106
+ os.makedirs(output_dir, exist_ok=True)
107
+ filepath = os.path.join(output_dir, filename)
108
+ image.save(filepath, "JPEG", quality=80)
109
+ return filepath