update gradio mistake

app.py

@@ -12,12 +12,12 @@ pipe = StableDiffusionPipeline.from_single_file(
     use_safetensors=True,
 )
 
-num_images_per_prompt = 2
+num_images_per_prompt = 4
 prompt = 'Modern Elegance:Explore the seamless blend of sleek lines, minimalist aesthetics, and cutting-edge design in modern interior decor'
 
 negative_prompt = '(nsfw:1.3),(Nude:1.3),(Naked:1.3),low quality, blurry, bad anatomy, worst quality, text, watermark, normal quality, ugly, signature, lowres, deformed, disfigured, cropped, jpeg artifacts, error, \
                     mutation, logo, watermark,text, logo,contact, error, blurry, cropped, username, artist name, (worst quality, low quality:1.4),monochrome,'
-def inference(prompt):
+def inference(prompt,ref_image):
     image = pipe(
         prompt=prompt, 
         negative_prompt=negative_prompt,
@@ -32,17 +32,18 @@ def inference(prompt):
     ).images
     return image
 
-
-INPUT_HEIGHT = 512
-INPUT_WIDTH = 512
 with gr.Row():
     input_prompt = gr.Textbox(placeholder="输入你对室内设计的要求，以便 AI 能够更好地满足你的需求。", label="要求")
 with gr.Row():
-    ref_image = gr.Image(label="参考图片").style(height=INPUT_HEIGHT, width=INPUT_WIDTH)
-    result_image = gr.Gallery().style(columns=3,height=INPUT_HEIGHT, width=INPUT_WIDTH)
+    ref_image = gr.Image(height=512, width=512, label="参考图片")
+    
+    result_image = gr.Gallery(label="可能满足你需求的室内设计图：", 
+                                    columns=3, 
+                                    height="auto", 
+                                    object_fit="contain")
 demo = gr.Interface(
     fn=inference,
-    inputs=[input_prompt],
+    inputs=[input_prompt,ref_image],
     outputs=[result_image]
 )