Nadine Rueegg commited on
Commit
2a77bc9
β€’
1 Parent(s): 4ff797f

update packages and requirements

Browse files
Files changed (3) hide show
  1. README.md +8 -7
  2. app.py +1 -1
  3. gradio_demo/barc_demo_v3.py +9 -0
README.md CHANGED
@@ -1,12 +1,13 @@
1
  ---
2
- title: Barc Gradio
3
- emoji: πŸ”₯
4
- colorFrom: red
5
- colorTo: yellow
6
  sdk: gradio
7
- sdk_version: 3.0.24
8
- app_file: app.py
9
  pinned: false
 
10
  ---
11
 
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
  ---
2
+ title: BARC
3
+ emoji: πŸ§πŸ½β€β™€οΈπŸ§πŸ»πŸ§πŸ½β€β™‚οΈΒ 
4
+ colorFrom: pink
5
+ colorTo: green
6
  sdk: gradio
7
+ sdk_version: 3.0.2
8
+ app_file: ./gradio_demo/barc_demo_v3.py
9
  pinned: false
10
+ python_version: 3.7.6
11
  ---
12
 
13
+ Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
app.py CHANGED
@@ -218,7 +218,7 @@ def run_complete_inference(input_image):
218
  # see: https://huggingface.co/spaces/radames/PIFu-Clothed-Human-Digitization/blob/main/PIFu/spaces.py
219
 
220
  description = '''
221
- # BARC
222
 
223
  #### Project Page
224
  * https://barc.is.tue.mpg.de/
 
218
  # see: https://huggingface.co/spaces/radames/PIFu-Clothed-Human-Digitization/blob/main/PIFu/spaces.py
219
 
220
  description = '''
221
+ # BARC (old)
222
 
223
  #### Project Page
224
  * https://barc.is.tue.mpg.de/
gradio_demo/barc_demo_v3.py CHANGED
@@ -1,5 +1,14 @@
1
  # python gradio_demo/barc_demo_v3.py
2
 
 
 
 
 
 
 
 
 
 
3
  import numpy as np
4
  import os
5
  import glob
 
1
  # python gradio_demo/barc_demo_v3.py
2
 
3
+ import os
4
+ os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
5
+ os.environ["CUDA_VISIBLE_DEVICES"]="0"
6
+ try:
7
+ # 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")
8
+ 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")
9
+ except Exception as e:
10
+ print(e)
11
+
12
  import numpy as np
13
  import os
14
  import glob