Hugging Face's logo

Spaces:
akhaliq
/
Mask2Former
Runtime error

App Files Community
1
akhaliq HF staff commited on
Commit
106c46c
•
1 Parent(s): 2220924

Update app.py

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