Spaces:
Runtime error
Runtime error
Nadine Rueegg
commited on
Commit
·
7b976bf
1
Parent(s):
c90d71a
adjust script
Browse files- scripts/gradio_demo.py +2 -2
scripts/gradio_demo.py
CHANGED
@@ -8,14 +8,14 @@
|
|
8 |
# for all images from the folder datasets/test_image_crops:
|
9 |
# python scripts/gradio.py --workers 12 --config refinement_cfg_test_withvertexwisegc_csaddnonflat_crops.yaml --model-file-complete=cvpr23_dm39dnnv3barcv2b_refwithgcpervertisflat0morestanding0/checkpoint.pth.tar -s ttopt_vtest2
|
10 |
|
11 |
-
|
12 |
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
|
13 |
os.environ["CUDA_VISIBLE_DEVICES"]="0"
|
14 |
try:
|
15 |
# os.system("pip install --upgrade torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html")
|
16 |
os.system("pip install --upgrade torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/cu101/torch_stable.html")
|
17 |
except Exception as e:
|
18 |
-
print(e)
|
19 |
|
20 |
import argparse
|
21 |
import os.path
|
|
|
8 |
# for all images from the folder datasets/test_image_crops:
|
9 |
# python scripts/gradio.py --workers 12 --config refinement_cfg_test_withvertexwisegc_csaddnonflat_crops.yaml --model-file-complete=cvpr23_dm39dnnv3barcv2b_refwithgcpervertisflat0morestanding0/checkpoint.pth.tar -s ttopt_vtest2
|
10 |
|
11 |
+
import os
|
12 |
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
|
13 |
os.environ["CUDA_VISIBLE_DEVICES"]="0"
|
14 |
try:
|
15 |
# os.system("pip install --upgrade torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html")
|
16 |
os.system("pip install --upgrade torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/cu101/torch_stable.html")
|
17 |
except Exception as e:
|
18 |
+
print(e)
|
19 |
|
20 |
import argparse
|
21 |
import os.path
|