Update app.py

app.py

@@ -11,6 +11,32 @@ from image_segmenter import ImageSegmenter
 from monocular_depth_estimator import MonocularDepthEstimator
 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
+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("cpu")
+
+
 # params
 CANCEL_PROCESSING = False
 
@@ -186,4 +212,4 @@ if __name__ == "__main__":
         conf_thres_vid.change(update_confidence_threshold, conf_thres_vid, [])       
 
 
-    my_app.queue(concurrency_count=5, max_size=20).launch() 
+    my_app.queue(max_size=20).launch()