Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,6 +12,13 @@ from werkzeug.utils import secure_filename
|
|
| 12 |
import warnings
|
| 13 |
warnings.filterwarnings("ignore")
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
# モデルと設定の初期化
|
| 16 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 17 |
|
|
|
|
| 12 |
import warnings
|
| 13 |
warnings.filterwarnings("ignore")
|
| 14 |
|
| 15 |
+
os.system("git clone https://github.com/xuebinqin/DIS")
|
| 16 |
+
os.system("mv DIS/IS-Net/* .")
|
| 17 |
+
|
| 18 |
+
# project imports
|
| 19 |
+
from data_loader_cache import normalize, im_reader, im_preprocess
|
| 20 |
+
from models import *
|
| 21 |
+
|
| 22 |
# モデルと設定の初期化
|
| 23 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 24 |
|