Create app.py

app.py

@@ -0,0 +1,30 @@
+import torch
+from diffusers import StableDiffusionPipeline
+import imageio
+import os
+
+def generate_images(prompt, num_images=5, output_folder="generated_frames"):
+    # Ensure output folder exists
+    os.makedirs(output_folder, exist_ok=True)
+    
+    # Load model (CPU only)
+    pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
+    pipe.to("cpu")  # Force to run on CPU
+    
+    images = []
+    for i in range(num_images):
+        image = pipe(prompt).images[0]  # Generate image
+        image_path = os.path.join(output_folder, f"frame_{i}.png")
+        image.save(image_path)
+        images.append(imageio.imread(image_path))
+    
+    # Create GIF animation
+    gif_path = os.path.join(output_folder, "animation.gif")
+    imageio.mimsave(gif_path, images, duration=0.5)
+    
+    print(f"Animation saved at {gif_path}")
+    return gif_path
+
+if __name__ == "__main__":
+    prompt = input("Enter your prompt: ")
+    generate_images(prompt)