Use CPU

app.py

@@ -19,16 +19,16 @@ ffmpeg_path = shutil.which("ffmpeg")
 mediapy.set_ffmpeg(ffmpeg_path)
 
 # Loading model
-device = torch.device("cuda")
+device = torch.device("cpu")
 vqvae = mlflow.pytorch.load_model(
     "./trained_models/vae/",
-    # map_location=device,
+    map_location=device,
 )
 vqvae.eval()
 
 diffusion = mlflow.pytorch.load_model(
     "./trained_models/ddpm/",
-    # map_location=device,
+    map_location=device,
 )
 diffusion.eval()