neew

app.py

@@ -20,13 +20,13 @@ pipe = DiffusionPipeline.from_pretrained(
     torch_dtype=torch_dtype,
     trust_remote_code=True,
     use_auth_token=token
-).to("cuda" if torch.cuda.is_available() else "cpu")
+).to("cuda")
 
 # Enable memory-saving features
 pipe.enable_attention_slicing()
 
 # Generation function
-def generate_video(image, prompt, num_frames=16, steps=50, guidance_scale=7.5):
+def generate_video(image, prompt, num_frames=16, steps=25, guidance_scale=7.5):
     output = pipe(
         prompt=prompt,
         image=image,