Update app.py

app.py

@@ -28,8 +28,15 @@ def clear_cache(transformer):
 
 # Define the Gradio interface
 @spaces.GPU()
-def single_condition_generate_image(prompt, spatial_img, height, width, seed, control_type, use_zero_init, zero_steps):
+def single_condition_generate_image(spatial_img, , , ):
     # Set the control type
+    prompt = "Ghibli Studio style, Charming hand-drawn anime-style illustration"
+    use_zero_init = False
+    zero_steps = 1
+    control_type = "Ghibli"
+    height = 768
+    width = 768
+    seed = 42
     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)
@@ -62,14 +69,7 @@ with gr.Blocks() as demo:
     with gr.Tab("Ghibli Condition Generation"):
         with gr.Row():
             with gr.Column():
-                prompt = gr.Textbox(label="Prompt", value="Ghibli Studio style, Charming hand-drawn anime-style illustration")
                 spatial_img = gr.Image(label="Ghibli Image", type="pil")  # 上传图像文件
-                height = gr.Slider(minimum=256, maximum=1024, step=64, label="Height", value=768)
-                width = gr.Slider(minimum=256, maximum=1024, step=64, label="Width", value=768)
-                seed = gr.Number(label="Seed", value=42)
-                control_type = gr.Dropdown(choices=control_types, label="Control Type")
-                use_zero_init = gr.Checkbox(label="Use CFG zero star", value=False)
-                zero_steps = gr.Number(label="Zero Init Steps", value=1)
                 single_generate_btn = gr.Button("Generate Image")
                 
             with gr.Column():
@@ -78,7 +78,7 @@ with gr.Blocks() as demo:
     # Link the buttons to the functions
     single_generate_btn.click(
         single_condition_generate_image,
-        inputs=[prompt, spatial_img, height, width, seed, control_type, use_zero_init, zero_steps],
+        inputs=[spatial_img],
         outputs=single_output_image
     )