Upload app.py with huggingface_hub

app.py

@@ -114,7 +114,7 @@ def get_game(name: str):
         samp_cfg.num_steps_denoising = DENOISING_STEPS
         sampler = DiffusionSampler(denoiser, samp_cfg)
 
-        seeds = [Episode.load(p) for p in sorted(glob.glob(str(d / "seeds" / "*.pt")))]
+        seeds = [Episode.load(p, map_location=DEVICE) for p in sorted(glob.glob(str(d / "seeds" / "*.pt")))]
         assert seeds, f"no seed episodes in {d/'seeds'}"
 
         noop = action_names.index("noop") if "noop" in action_names else 0