Spaces:
Sleeping
Sleeping
do not merge
#1
by
not-lain
- opened
app.py
CHANGED
|
@@ -19,7 +19,7 @@ os.system("mv DIS/IS-Net/* .")
|
|
| 19 |
from data_loader_cache import normalize, im_reader, im_preprocess
|
| 20 |
from models import *
|
| 21 |
|
| 22 |
-
#Helpers
|
| 23 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 24 |
|
| 25 |
# Download official weights
|
|
|
|
| 19 |
from data_loader_cache import normalize, im_reader, im_preprocess
|
| 20 |
from models import *
|
| 21 |
|
| 22 |
+
# Helpers
|
| 23 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 24 |
|
| 25 |
# Download official weights
|