Update app.py

app.py

@@ -3,7 +3,9 @@ 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.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/")