Update app.py

app.py

@@ -5,12 +5,15 @@ api_key = "your_huggingface_api_key"
 
 # Load the Stable Diffusion pipeline with the API key
 pipeline = StableDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-v1-5-base",
+    "CompVis/stable-diffusion-v1-4",
     use_auth_token=api_key
 )
 
+# Move the pipeline to GPU if available
+pipeline.to("cuda" if torch.cuda.is_available() else "cpu")
+
 # Generate an image
-prompt = "A beautiful futuristic cityscape at night"
+prompt = "A serene lake surrounded by mountains during sunset"
 image = pipeline(prompt).images[0]
 
 # Save the generated image