num image

app.py

@@ -112,21 +112,23 @@ threading.Thread(target=delete_old_files, daemon=True).start()
 
 
 @spaces.GPU(duration=180)  # [uncomment to use ZeroGPU]
-def infer(prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
+def infer(prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, num_images,
           progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, 65536)
 
-    image = pipe(
+    images = pipe(
         prompt=prompt,
         guidance_scale=guidance_scale,
-        num_images_per_prompt=1,
+        num_images_per_prompt=num_images,  # 生成 num_images 张图
         num_inference_steps=num_inference_steps,
         width=width,
         height=height,
         generator=torch.Generator().manual_seed(seed)
-    ).images[0]
-    return image, seed
+    ).images  # 获取生成的图片列表
+
+    return images, seed
+
 
 
 examples = [
@@ -247,7 +249,7 @@ with gr.Blocks() as demo:
                         scale=3
                     )
                     ex_img = gr.Image(
-                        value=f"../img/img_{i + 1}.png",
+                        value=f"img/img_{i + 1}.png",
                         label="Effect",
                         interactive=False,
                         height=130,