Update app.py

app.py

@@ -1,12 +1,11 @@
 import gradio as gr
 from diffusers import DiffusionPipeline
 
-def generate_image(btn):
-    if btn:
-        pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
-        pipe = pipeline.to("cuda")
-        image = pipe().images[0]
-        return image
+def generate_image(width):
+    pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
+    pipe = pipeline.to("cuda")
+    image = pipe(image_size=width).images[0]
+    return image
 
 app = gr.Interface(
     fn=generate_image,
@@ -14,6 +13,4 @@ app = gr.Interface(
     outputs="image",
     title="Cat Image Generator",
     description="Click the button to generate a cute cat image."
-)
-
-app.launch()
+).launch()