Add device

Image3DProcessor.py

@@ -26,8 +26,7 @@ from gaussian_renderer import render_predicted
 
 class Image3DProcessor:
     def __init__(self, model_cfg_path, model_repo_id, model_filename):
-        Image3DProcessor._setup_done = True  # Mark setup as done
-        
+        self.device = torch.device("cuda" if self.use_cuda else "cpu")
         # Load model configuration
         self.model_cfg = OmegaConf.load(model_cfg_path)