update

app.py

@@ -165,7 +165,7 @@ def save_video(frames, path="output.mp4", fps=10):
 def run(cfg: DictConfig):
 
     algo = run_local(cfg)
-    # algo.to(device)
+    algo.to(device)
     
     actions = torch.zeros((1, 25))
     poses = torch.zeros((1, 5))
@@ -174,13 +174,11 @@ def run(cfg: DictConfig):
 
     @spaces.GPU()
     def run_interactive(algo, first_frame, action, first_pose, curr_frame, device):
-        algo = algo.to(device)
         new_frame = algo.interactive(first_frame,
                                         action,
                                         first_pose, 
                                         curr_frame, 
                                         device=device)
-        algo = algo.to("cpu")
         return algo, new_frame
 
     def set_denoising_steps(denoising_steps, sampling_timesteps_state):