Update app.py

app.py

@@ -1,13 +1,17 @@
 import spaces
 import os
-os.system("pip install git+https://github.com/facebookresearch/detectron2.git")
-os.system("git clone https://github.com/Visual-AI/Mr.DETR.git MrDETR && cd MrDETR && rm -f requirements.txt && cd ..")
-# os.system("cp multi_scale_deform_attn.py MrDETR/detrex/layers/ && cd MrDETR && pip install . & cd ..")
-# os.system("cd MrDETR && export CUDA_HOME=$(dirname $(dirname $(which nvcc))) && pip install . & cd ..")
-os.environ["CUDA_HOME"] = os.environ.get("CUDA_PATH")
-os.system("cd MrDETR && pip install . & cd ..")
-import sys 
-sys.path.append("MrDETR/")
+
+@spaces.GPU
+def install_setup():
+    os.system("pip install git+https://github.com/facebookresearch/detectron2.git")
+    os.system("git clone https://github.com/Visual-AI/Mr.DETR.git MrDETR && cd MrDETR && rm -f requirements.txt && cd ..")
+    # os.system("cp multi_scale_deform_attn.py MrDETR/detrex/layers/ && cd MrDETR && pip install . & cd ..")
+    # os.system("cd MrDETR && export CUDA_HOME=$(dirname $(dirname $(which nvcc))) && pip install . & cd ..")
+    os.environ["CUDA_HOME"] = os.environ.get("CUDA_PATH")
+    os.system("cd MrDETR && pip install . & cd ..")
+    import sys 
+    sys.path.append("MrDETR/")
+install_setup()
 
 import gradio as gr
 from demo.predictors import VisualizationDemo