install detectron2 from script

app.py

@@ -1,9 +1,16 @@
+import os
 import streamlit as st
+
+try:
+    import detectron2
+except:
+    with st.spinner("Installing detectron2"):
+        os.system("pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.10/index.html")
+
 import cv2
 import sys
 import argparse
 import numpy as np
-import os
 import json
 import torch
 import torch.nn.functional as F
@@ -13,6 +20,7 @@ from collections import OrderedDict
 from scipy import spatial
 import matplotlib.pyplot as plt
 
+
 from detectron2.engine import DefaultPredictor
 from detectron2.utils.visualizer import Visualizer
 from detectron2.config import get_cfg