test ControlNet with a random generated dummy img

app.py

@@ -52,6 +52,7 @@ import torch
 from diffusers import StableDiffusionPipeline, ControlNetModel, StableDiffusionControlNetPipeline
 from PIL import Image
 import gradio as gr
+import numpy as np
 
 # 自动选择设备
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -74,7 +75,8 @@ pipe.enable_attention_slicing()
 
 # 文生图生成函数
 def generate_image(prompt, num_steps=20, height=256, width=256):
-    image = pipe(prompt, num_inference_steps=num_steps, height=height, width=width).images[0]
+    dummy_image = Image.fromarray(np.zeros((height, width, 3), dtype=np.uint8))
+    image = pipe(prompt, num_inference_steps=num_steps, height=height, width=width,image=dummy_image).images[0]
     return image
 
 # Gradio 界面