Update app.py

app.py

@@ -1,5 +1,17 @@
 #!/usr/bin/env python3
+import importlib.util
+import os
+import sys
+
+# Check if detectron2 is installed
+if importlib.util.find_spec("detectron2") is None:
+    print("Installing PyTorch and Detectron2...")
+    os.system("pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu")
+    os.system("pip install git+https://github.com/facebookresearch/detectron2.git")
+    print("Installation complete!")
+
 # -*- coding: utf-8 -*-
+
 import os
 import sys
 import time