Spaces:
Sleeping
Sleeping
Commit
·
41517aa
1
Parent(s):
945554c
add new and remove models
Browse files
main.py
CHANGED
|
@@ -33,16 +33,26 @@ async def verify_api_key(request, call_next):
|
|
| 33 |
|
| 34 |
|
| 35 |
models_locations = [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
{
|
| 37 |
"repo_id": "TamisAI/category-lamp",
|
| 38 |
-
"subfolder": "maison-
|
| 39 |
-
"filename": "
|
| 40 |
},
|
| 41 |
{
|
| 42 |
"repo_id": "TamisAI/category-lamp",
|
| 43 |
"subfolder": "maison-charles/corail-152-0001-32-256",
|
| 44 |
"filename": "maison-charles-corail.pth",
|
| 45 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
]
|
| 47 |
|
| 48 |
device = torch.device("cpu")
|
|
|
|
| 33 |
|
| 34 |
|
| 35 |
models_locations = [
|
| 36 |
+
# {
|
| 37 |
+
# "repo_id": "TamisAI/category-lamp",
|
| 38 |
+
# "subfolder": "maison-jansen/palmtree-152-0005-32-256",
|
| 39 |
+
# "filename": "palmtree-jansen.pth",
|
| 40 |
+
# },
|
| 41 |
{
|
| 42 |
"repo_id": "TamisAI/category-lamp",
|
| 43 |
+
"subfolder": "maison-charles/corail-152-0001-32-256-old",
|
| 44 |
+
"filename": "maison-charles-corail.pth",
|
| 45 |
},
|
| 46 |
{
|
| 47 |
"repo_id": "TamisAI/category-lamp",
|
| 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-50-0001-32-256",
|
| 54 |
+
"filename": "maison-charles-corail.pth",
|
| 55 |
+
},
|
| 56 |
]
|
| 57 |
|
| 58 |
device = torch.device("cpu")
|