update README

app.py

@@ -175,7 +175,10 @@ def run(cfg: DictConfig):
 
     algo = run_local(cfg)
     algo.to(device)
-    
+
+    device = next(algo.parameters()).device
+    print("algo:", device)
+
     actions = torch.zeros((1, 25))
     poses = torch.zeros((1, 5))
     
@@ -183,6 +186,9 @@ def run(cfg: DictConfig):
 
     runner = InteractiveRunner(algo)
 
+    device = next(runner.algo.parameters()).device
+    print("runner.algo:", device)
+
     # @spaces.GPU()
     # def run_interactive(first_frame, action, first_pose, curr_frame, device):
     #     global algo