Update app.py

app.py

@@ -14,26 +14,12 @@ from point_cloud_generator import display_pcd
 import os
 import torch
 
-# Disable NVML initialization to avoid errors in Hugging Face Spaces
-os.environ["CUDA_LAUNCH_BLOCKING"] = "1"
-os.environ["TORCH_CUDA_ARCH_LIST"] = ""
-os.environ["CUDA_VISIBLE_DEVICES"] = "0" if torch.cuda.is_available() else "-1"
-
-
-# Ensure CUDA is properly initialized
+# Let ZeroGPU handle initialization
 try:
-    if torch.cuda.is_available():
-        print(f"✅ CUDA is available: {torch.cuda.get_device_name(0)}")
-        torch.cuda.init()
-        torch.cuda.empty_cache()
-        device = torch.device("cuda")
-    else:
-        print("❌ No CUDA available. Falling back to CPU.")
-        os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
-        device = torch.device("cpu")
-except RuntimeError as e:
-    print(f"🚨 CUDA initialization failed: {e}. Switching to CPU mode.")
-    os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
+    device = torch.device("cuda" if spaces.is_gpu_available() else "cpu")
+    print(f"🔹 Running inference on {device}")
+except Exception as e:
+    print(f"❌ ZeroGPU failed to initialize: {e}")
     device = torch.device("cpu")