Update app.py

app.py

@@ -1,17 +1,20 @@
-from transformers import pipeline
+from diffusers import StableDiffusionPipeline
 
-# Define your API key
+# Replace with your Hugging Face API key
 api_key = "your_huggingface_api_key"
 
-# Initialize the pipeline using the API key
-generator = pipeline(
-    "image-generation",
-    model="CompVis/stable-diffusion-v1-4-original",
+# Load the Stable Diffusion pipeline
+pipeline = StableDiffusionPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-v1-5-base",
     use_auth_token=api_key
 )
 
-# Generate an image
-result = generator("A futuristic city")
+# Generate an image from a text prompt
+prompt = "A serene mountain landscape at sunset with a lake reflecting the sky"
+image = pipeline(prompt).images[0]
 
-# Display the generated image
-result[0]["image"].show()
+# Save the generated image to a file
+image.save("generated_image_v1_5.png")
+
+# Optionally display the image if your environment supports it
+image.show()