Update app.py

app.py

@@ -14,6 +14,10 @@ pipeline = StableDiffusionPipeline.from_pretrained(
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipeline.to(device)
 
+# Reduce the image resolution for faster generation
+pipeline.set_format("jpeg")
+pipeline.set_resolution(512)  # Reduce resolution (e.g., 512x512)
+
 # Generate an image
 prompt = "A serene lake surrounded by mountains during sunset"
 image = pipeline(prompt).images[0]