Update app.py

app.py

@@ -2,11 +2,16 @@ from diffusers import StableDiffusionPipeline
 from PIL import Image
 import imageio
 import gradio as gr
+import torch
+
+# Dynamic device and data type allocation
+device = "cuda" if torch.cuda.is_available() else "cpu"
+dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
 # Load Stable Diffusion model
-model_id = "runwayml/stable-diffusion-v1-5"  # Or your custom fine-tuned model
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype="float16", safety_checker=None)
-pipe.to("cuda")  # Use GPU if available
+model_id = "runwayml/stable-diffusion-v1-5"  # Replace with your desired model ID
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=dtype, safety_checker=None)
+pipe.to(device)
 
 def generate_frames(prompt, num_frames=5, style="realistic"):
     """