Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 from diffusers import DiffusionPipeline
 
-# Load the diffusion pipeline
-pipe = DiffusionPipeline.from_pretrained("ByteDance/AnimateDiff-Lightning")
+# Load the Stable Diffusion pipeline
+pipe = DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v-1-4")
 
 def generate_image(prompt):
     # Generate the image from the prompt
@@ -14,8 +14,8 @@ iface = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(label="Enter your prompt", placeholder="e.g. Astronaut in a jungle"),
     outputs=gr.Image(label="Generated Image"),
-    title="Diffusion Pipeline Image Generator",
-    description="Generate images using a diffusion pipeline based on your prompts."
+    title="Stable Diffusion Image Generator",
+    description="Generate images using Stable Diffusion based on your prompts."
 )
 
 if __name__ == "__main__":