Spaces:
Sleeping
Sleeping
Nunzio commited on
Commit ·
5cfe5ea
1
Parent(s): ef2b3d9
TOMMASO TI AMO
Browse files
app.py
CHANGED
|
@@ -13,8 +13,8 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
| 13 |
|
| 14 |
|
| 15 |
MODELS = {
|
| 16 |
-
"
|
| 17 |
-
"
|
| 18 |
}
|
| 19 |
|
| 20 |
|
|
|
|
| 13 |
|
| 14 |
|
| 15 |
MODELS = {
|
| 16 |
+
"BISENET": loadBiSeNet(device),
|
| 17 |
+
"BISENETV2": loadBiSeNetV2(device)
|
| 18 |
}
|
| 19 |
|
| 20 |
|