update

app.py

@@ -22,12 +22,13 @@ pruned_pipe.unet = torch.load(
 )
 pruned_pipe = pruned_pipe.to(device)
 
+
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 
 @spaces.GPU  # [uncomment to use ZeroGPU]
-def generate_images(prompt, seed, steps, pipe, pruned_pipe):
+def generate_images(prompt, seed, steps):
     # Run the model and return images directly
     g_cpu = torch.Generator("cuda").manual_seed(seed)
     original_image = pipe(prompt=prompt, generator=g_cpu, num_inference_steps=steps).images[0]
@@ -49,13 +50,11 @@ css = """
 }
 """
 header = """
-# 🌱 Text-to-Image Generation with EcoDiff Pruned SD-XL (20% Pruning Ratio)
-# Under Construction!!!
+# 🌱 EcoDiff Pruned SD-XL (20% Pruning Ratio)
 
 <div style="text-align: center; display: flex; justify-content: left; gap: 5px;">
 <a href="https://arxiv.org/abs/2412.02852"><img src="https://img.shields.io/badge/ariXv-Paper-A42C25.svg" alt="arXiv"></a>
 <a href="https://huggingface.co/zhangyang-0123/EcoDiffPrunedModels"><img src="https://img.shields.io/badge/🤗-Model-ffbd45.svg" alt="HuggingFace"></a>
-<a href="https://github.com/YaNgZhAnG-V5/EcoDiff"><img src="https://img.shields.io/badge/GitHub-Code-blue.svg?logo=github&" alt="GitHub"></a>
 </div>
 """
 
@@ -97,8 +96,6 @@ with gr.Blocks(css=css) as demo:
             prompt,
             seed,
             steps,
-            pipe,
-            pipe,
         ],
         outputs=[original_output, ecodiff_output],
     )