Update app.py

app.py

@@ -28,14 +28,13 @@ def clear_cache(transformer):
 
 # Define the Gradio interface
 @spaces.GPU()
-def single_condition_generate_image(prompt, subject_img, spatial_img, height, width, seed, control_type):
+def single_condition_generate_image(prompt, spatial_img, height, width, seed, control_type):
     # Set the control type
     if control_type == "Ghibli":
         lora_path = os.path.join(lora_base_path, "Ghibli.safetensors")
     set_single_lora(pipe.transformer, lora_path, lora_weights=[1], cond_size=512)
     
     # Process the image
-    subject_imgs = [subject_img] if subject_img else []
     spatial_imgs = [spatial_img] if spatial_img else []
     image = pipe(
         prompt,
@@ -45,7 +44,7 @@ def single_condition_generate_image(prompt, subject_img, spatial_img, height, wi
         num_inference_steps=25,
         max_sequence_length=512,
         generator=torch.Generator("cpu").manual_seed(seed), 
-        subject_images=subject_imgs,
+        subject_images=[],
         spatial_images=spatial_imgs,
         cond_size=512,
     ).images[0]
@@ -84,7 +83,7 @@ with gr.Blocks() as demo:
         # Add examples for Single Condition Generation
         gr.Examples(
             examples=single_examples,
-            inputs=[prompt, None, spatial_img, height, width, seed, control_type],
+            inputs=[prompt, spatial_img, height, width, seed, control_type],
             outputs=single_output_image,
             fn=single_condition_generate_image,
             cache_examples=False,  # 缓存示例结果以加快加载速度
@@ -94,7 +93,7 @@ with gr.Blocks() as demo:
     # Link the buttons to the functions
     single_generate_btn.click(
         single_condition_generate_image,
-        inputs=[prompt, None, spatial_img, height, width, seed, control_type],
+        inputs=[prompt, spatial_img, height, width, seed, control_type],
         outputs=single_output_image
     )