Spaces:
Runtime error
Runtime error
Jitesh Jain
commited on
Commit
·
bd11a0f
1
Parent(s):
cb54a13
Fix packages
Browse files- app.py +1 -0
- requirements.txt +3 -2
- setup.sh +1 -2
- test_512/person512.png +0 -0
app.py
CHANGED
|
@@ -10,6 +10,7 @@ import numpy as np
|
|
| 10 |
import torch
|
| 11 |
import legacy
|
| 12 |
import paddlehub as hub
|
|
|
|
| 13 |
|
| 14 |
u2net = hub.Module(name='U2Net')
|
| 15 |
|
|
|
|
| 10 |
import torch
|
| 11 |
import legacy
|
| 12 |
import paddlehub as hub
|
| 13 |
+
import cv2
|
| 14 |
|
| 15 |
u2net = hub.Module(name='U2Net')
|
| 16 |
|
requirements.txt
CHANGED
|
@@ -1,4 +1,7 @@
|
|
| 1 |
icecream
|
|
|
|
|
|
|
|
|
|
| 2 |
psutil
|
| 3 |
click
|
| 4 |
requests
|
|
@@ -22,5 +25,3 @@ kornia==0.5.0
|
|
| 22 |
gradio
|
| 23 |
ipython
|
| 24 |
Jinja2
|
| 25 |
-
paddlepaddle
|
| 26 |
-
paddlehub
|
|
|
|
| 1 |
icecream
|
| 2 |
+
opencv-python-headless
|
| 3 |
+
paddlepaddle
|
| 4 |
+
paddlehub
|
| 5 |
psutil
|
| 6 |
click
|
| 7 |
requests
|
|
|
|
| 25 |
gradio
|
| 26 |
ipython
|
| 27 |
Jinja2
|
|
|
|
|
|
setup.sh
CHANGED
|
@@ -1,3 +1,2 @@
|
|
| 1 |
#!/bin/sh
|
| 2 |
-
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
|
| 3 |
-
pip install -r requirements.txt
|
|
|
|
| 1 |
#!/bin/sh
|
| 2 |
+
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
|
|
|
test_512/person512.png
CHANGED
|
|