Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
c542af4
1
Parent(s):
e7fd9ba
add new model
Browse files
main.py
CHANGED
|
@@ -48,6 +48,11 @@ models_locations = [
|
|
| 48 |
"subfolder": "maison-charles/corail-152-0001-32-256",
|
| 49 |
"filename": "maison-charles-corail.pth",
|
| 50 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
]
|
| 52 |
|
| 53 |
device = torch.device("cpu")
|
|
|
|
| 48 |
"subfolder": "maison-charles/corail-152-0001-32-256",
|
| 49 |
"filename": "maison-charles-corail.pth",
|
| 50 |
},
|
| 51 |
+
{
|
| 52 |
+
"repo_id": "TamisAI/category-lamp",
|
| 53 |
+
"subfolder": "maison-charles/corail-152-0001-32-256-lra",
|
| 54 |
+
"filename": "maison-charles-corail-lra.pth",
|
| 55 |
+
},
|
| 56 |
]
|
| 57 |
|
| 58 |
device = torch.device("cpu")
|