Nadine Rueegg commited on
Commit
7b976bf
1 Parent(s): c90d71a

adjust script

Browse files
Files changed (1) hide show
  1. 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
- '''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
 
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