Update app.py

app.py

@@ -263,9 +263,7 @@ def add_file(history, task_history, file, task_new, fig, query):
         coords        = ((torch.from_numpy(position_recon) + 0.5) * 64).int().contiguous()
         ss            = torch.zeros(1, 64, 64, 64, dtype=torch.long)
         ss[:, coords[:, 0], coords[:, 1], coords[:, 2]] = 1
-        print(ss)
-        print(ss.max(),ss.min())
-        token         = vqvae.Encode(ss.to(dtype=torch.float32).unsqueeze(0))
+        token         = vqvae.Encode(ss.to(dtype=torch.float32).unsqueeze(0).to("cuda"))
         token         = token[0].cpu().numpy().tolist()
         words         = token_to_words(token)
         fig           = make_pointcloud_figure(position_recon,rotate=True)