clear early cuda

app.py

@@ -847,8 +847,6 @@ if __name__ == '__main__':
     dpt_pack = None
     generated_object_map = {}
 
-    DEVICE = "cuda"
-
     # Prepare models
     ## Grounding SAM
     segmenter_id = "facebook/sam-vit-base"
@@ -866,10 +864,6 @@ if __name__ == '__main__':
     model_dir = 'HorizonRobotics/3D-Fixer'
     local_dir = 'checkpoints/3D-Fixer'
     snapshot_download(repo_id=model_dir, local_dir=local_dir)
-    # pipeline = ThreeDFixerPipeline.from_pretrained(
-    #     model_dir, compile=False
-    # )
-    # pipeline.to(device=DEVICE)
     ############## 3D-Fixer model
 
     rot = np.array([
@@ -884,6 +878,6 @@ if __name__ == '__main__':
         [0.0, 0.0, -1.0, 0.0],
         [0.0, 1.0, 0.0, 0.0],
         [0.0, 0.0, 0.0, 1.0],
-    ], dtype=torch.float32, device=DEVICE)
+    ], dtype=torch.float32, device='cpu')
 
     demo.launch(allowed_paths=[TMP_DIR, EXAMPLE_DIR])