Update app.py

app.py

@@ -1,26 +1,18 @@
 import os 
 import os
 
-os.system("wget https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64 -O cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb")
-os.system("sudo dpkg-i  cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb")
-os.system("sudo apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub")
-os.system("sudo apt-get update")
-os.system("sudo apt-get install cuda")
 
-os.system("nvcc --version")
-print(os.environ.get('CUDA_PATH'))
 os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
-os.system("git clone https://github.com/facebookresearch/Mask2Former.git")
+os.system("git clone https://github.com/AK391/Mask2Former.git")
 
 os.chdir("Mask2Former")
-os.system("pwd")
 os.system("pip install git+https://github.com/cocodataset/panopticapi.git")
-#os.chdir("mask2former/modeling/pixel_decoder/ops")
-#os.system("pwd")
+os.chdir("mask2former/modeling/pixel_decoder/ops")
+#s.system("pwd")
 
-#os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0"
-#os.environ["FORCE_CUDA"] = "1"
-#os.system("python setup.py build install")
+os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0"
+os.environ["FORCE_CUDA"] = "1"
+os.system("python setup.py build install")
 os.chdir("/home/user/app/Mask2Former/")
 os.system("pwd")
 import gradio as gr