akhaliq HF staff commited on
Commit
5a7a90e
1 Parent(s): ab5c947

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -9,7 +9,10 @@ os.system("pwd")
9
  os.system("pip install git+https://github.com/cocodataset/panopticapi.git")
10
  os.chdir("mask2former/modeling/pixel_decoder/ops")
11
  os.system("pwd")
12
- os.system("TORCH_CUDA_ARCH_LIST='8.0' FORCE_CUDA=1 python setup.py build install")
 
 
 
13
  os.chdir("/home/user/app/Mask2Former/")
14
  os.system("pwd")
15
  import gradio as gr
 
9
  os.system("pip install git+https://github.com/cocodataset/panopticapi.git")
10
  os.chdir("mask2former/modeling/pixel_decoder/ops")
11
  os.system("pwd")
12
+
13
+ os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0"
14
+ os.environ["FORCE_CUDA"] = "1"
15
+ os.system("python setup.py build install")
16
  os.chdir("/home/user/app/Mask2Former/")
17
  os.system("pwd")
18
  import gradio as gr