Update app.py

app.py

@@ -11,18 +11,37 @@ from image_segmenter import ImageSegmenter
 from monocular_depth_estimator import MonocularDepthEstimator
 from point_cloud_generator import display_pcd
 
-import os
-import torch
 
-# Let ZeroGPU handle initialization
-try:
-    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}")
+
+import subprocess
+
+# Check if CUDA is available
+def check_gpu():
+    try:
+        # Run nvidia-smi to check for GPU
+        gpu_check = subprocess.run(["nvidia-smi"], capture_output=True, text=True)
+        if "NVIDIA-SMI" in gpu_check.stdout:
+            print("✅ GPU is available!")
+            return True
+        else:
+            print("❌ No GPU detected!")
+            return False
+    except FileNotFoundError:
+        print("❌ No nvidia-smi found! GPU likely unavailable.")
+        return False
+
+# Set device based on GPU availability
+if check_gpu() and torch.cuda.is_available():
+    print(f"✅ CUDA available: {torch.cuda.get_device_name(0)}")
+    device = torch.device("cuda")
+    torch.cuda.empty_cache()
+else:
+    print("❌ No CUDA available. Falling back to CPU.")
+    os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
     device = torch.device("cpu")
 
 
+
 # params
 CANCEL_PROCESSING = False