Update app.py

app.py

@@ -1,12 +1,18 @@
-import os, sys, subprocess
+import torch, subprocess, sys
 
-# install detectron2 during space build
-if not os.path.exists("detectron2"):
-    subprocess.run(
-        ["pip", "install",
-         "git+https://github.com/facebookresearch/detectron2.git"],
-        check=True
-    )
+print("CUDA available:", torch.cuda.is_available())
+
+try:
+    import detectron2
+except:
+    print("Installing detectron2...")
+    subprocess.check_call([
+        sys.executable, "-m", "pip", "install",
+        "git+https://github.com/facebookresearch/detectron2.git"
+    ])
+    import detectron2
+
+print("Detectron2 ready")
 import gradio as gr
 import cv2
 import numpy as np