Update app.py

app.py

@@ -6,7 +6,7 @@ api_key = "your_huggingface_api_key"
 
 # Load the Stable Diffusion pipeline with the API key
 pipeline = StableDiffusionPipeline.from_pretrained(
-    "CompVis/stable-diffusion-v1-4",
+    "stabilityai/stable-diffusion-2",  # You can try using the newer and faster version of Stable Diffusion.
     use_auth_token=api_key
 )
 
@@ -14,13 +14,13 @@ 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)
+# Define a smaller resolution for faster generation
+height = 512  # You can try 512 or 256 to make it faster
+width = 512
 
 # Generate an image
 prompt = "A serene lake surrounded by mountains during sunset"
-image = pipeline(prompt).images[0]
+image = pipeline(prompt, height=height, width=width).images[0]
 
 # Save the generated image
 image.save("generated_image.png")