Update app.py

app.py

@@ -7,10 +7,12 @@ def generate_image(width):
     image = pipe(image_size=width).images[0]
     return image
 
-app = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Slider(label="Width", value=128, minimum=128, maximum=1024, step=2, info="The width of the image in pixels"),
-    outputs="image",
-    title="Cat Image Generator",
-    description="Click the button to generate a cute cat image."
-).launch()
+with gr.Blocks() as demo:
+    with gr.Row():
+        width_slider = gr.Slider(label="Width", value=128, minimum=128, maximum=1024, step=2, info="The width of the image in pixels")
+        output_image = gr.Image(label="Generated Image")
+
+    btn = gr.Button("Generate Image")
+    btn.click(fn=generate_image, inputs=width_slider, outputs=output_image)
+
+demo.launch()