Update app.py

将输出文件webp改为jpg

app.py

@@ -7,6 +7,7 @@ import numpy as np
 import random
 import spaces
 import torch
+import uuid
 from diffusers import Flux2Pipeline, Flux2Transformer2DModel
 from diffusers import BitsAndBytesConfig as DiffBitsAndBytesConfig
 import requests
@@ -15,6 +16,7 @@ import json
 import base64
 from huggingface_hub import InferenceClient
 
+
 subprocess.check_call([sys.executable, "-m", "pip", "install", "spaces==0.43.0"])
 
 dtype = torch.bfloat16
@@ -80,6 +82,14 @@ pipe.to(device)
 # flash-attn估计库估计更新了，导致冲突了，不使用预编译的了
 # spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/FLUX.2", variant="fa3")
 
+def save_image(image, output_dir):
+    filename = str(uuid.uuid4()) + ".jpg"
+    os.makedirs(output_dir, exist_ok=True)
+    filepath = os.path.join(output_dir, filename)
+    image.save(filepath, "JPEG", quality=80)
+    return filepath
+
+
 def image_to_data_uri(img):
     buffered = io.BytesIO()
     img.save(buffered, format="PNG")
@@ -187,7 +197,11 @@ def generate_image(prompt_embeds, image_list, width, height, num_inference_steps
         height=height,
         callback_on_step_end=callback_fn,
     ).images[0]
-    return image
+
+    path = save_image(image, "./outputs")
+    progress(1, desc="Complete")
+
+    return path
 
 def infer(prompt, input_images=None, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=50, guidance_scale=2.5, prompt_upsampling=False, progress=gr.Progress()):
     
@@ -339,7 +353,7 @@ with gr.Blocks() as demo:
                 
                 
             with gr.Column():
-                result = gr.Image(label="Result", show_label=False)
+                result = gr.Image(label="Result",type="filepath", show_label=False)
             
         
         gr.Examples(