Update app.py

app.py

@@ -136,7 +136,7 @@ def main(video_path, audio_path, progress=gr.Progress(track_tqdm=True)):
     unet, _ = UNet3DConditionModel.from_pretrained(
         OmegaConf.to_container(config.model),
         inference_ckpt_path,  # load checkpoint
-        device="cpu",
+        #device="cpu",
     )
 
     unet = unet.to(dtype=torch.float16)