Update app.py

app.py

@@ -1,4 +1,12 @@
+import os, sys, subprocess
 
+# install detectron2 during space build
+if not os.path.exists("detectron2"):
+    subprocess.run(
+        ["pip", "install",
+         "git+https://github.com/facebookresearch/detectron2.git"],
+        check=True
+    )
 import gradio as gr
 import cv2
 import numpy as np
@@ -6,7 +14,7 @@ from detectron2.config import get_cfg
 from detectron2.engine import DefaultPredictor
 from detectron2.utils.visualizer import Visualizer, ColorMode
 from detectron2.data import MetadataCatalog
-import os
+
 
 # Load model
 cfg = get_cfg()