Update app.py

app.py

@@ -104,7 +104,7 @@ def generate_video(
     if image is not None:
         kwargs["image"] = image.convert("RGB")
 
-    with torch.cuda.amp.autocast(dtype=pipe.transformer.dtype), torch.no_grad():
+    with torch.amp.autocast("cuda", dtype=pipe.transformer.dtype), torch.no_grad():
         video_frames = pipe(**kwargs)[0]
 
     os.makedirs("gradio_videos", exist_ok=True)