Spaces:
Sleeping
Sleeping
Commit
·
4b470c5
1
Parent(s):
c542af4
update api with new models
Browse files
main.py
CHANGED
|
@@ -53,6 +53,16 @@ models_locations = [
|
|
| 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")
|
|
|
|
| 53 |
"subfolder": "maison-charles/corail-152-0001-32-256-lra",
|
| 54 |
"filename": "maison-charles-corail-lra.pth",
|
| 55 |
},
|
| 56 |
+
{
|
| 57 |
+
"repo_id": "TamisAI/category-lamp",
|
| 58 |
+
"subfolder": "maison-charles/corail-152-0001-32-256-ona",
|
| 59 |
+
"filename": "maison-charles-corail-ona.pth",
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"repo_id": "TamisAI/category-lamp",
|
| 63 |
+
"subfolder": "maison-charles/corail-152-0001-32-256-ona-ta",
|
| 64 |
+
"filename": "maison-charles-corail-ona-ta.pth",
|
| 65 |
+
},
|
| 66 |
]
|
| 67 |
|
| 68 |
device = torch.device("cpu")
|