Spaces:
Sleeping
Sleeping
add rvc models
Browse files
api.py
CHANGED
|
@@ -18,11 +18,26 @@ app = FastAPI()
|
|
| 18 |
preload()
|
| 19 |
|
| 20 |
path_models = [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
{
|
| 22 |
"name": "zeta",
|
| 23 |
-
"
|
| 24 |
-
"
|
| 25 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
},
|
| 27 |
]
|
| 28 |
|
|
|
|
| 18 |
preload()
|
| 19 |
|
| 20 |
path_models = [
|
| 21 |
+
# {
|
| 22 |
+
# "name": "zeta",
|
| 23 |
+
# "label": "Zeta",
|
| 24 |
+
# "ckpt_path": "weights/zet_test1.pth",
|
| 25 |
+
# "index_path": "weights/zet_test1.0.index"
|
| 26 |
+
# },
|
| 27 |
{
|
| 28 |
"name": "zeta",
|
| 29 |
+
"ckpt_path": "zeta.pth",
|
| 30 |
+
"index_path": "added_IVF409_Flat_nprobe_1.index"
|
| 31 |
+
},
|
| 32 |
+
{
|
| 33 |
+
"name": "kobov2",
|
| 34 |
+
"ckpt_path": "kobov2.pth",
|
| 35 |
+
"index_path": "added_IVF454_Flat_nprobe_1_kobov2_v2.index"
|
| 36 |
+
},
|
| 37 |
+
{
|
| 38 |
+
"name": "chamber",
|
| 39 |
+
"ckpt_path": "Chamber.pth",
|
| 40 |
+
"index_path": "added_IVF746_Flat_nprobe_1_Chamber_v2.index"
|
| 41 |
},
|
| 42 |
]
|
| 43 |
|
weights/Chamber.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c9f78ab2f90a67068e379cb26325b18eb41c9c381dea460531069fd0973e29bc
|
| 3 |
+
size 55222703
|
weights/added_IVF409_Flat_nprobe_1.index
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1ae945f03667899e790a0cb745281764a598f74323cb3bd5ef58677b34a3344b
|
| 3 |
+
size 16919779
|
weights/added_IVF454_Flat_nprobe_1_kobov2_v2.index
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:81ad383bd13b46b7a0c3b526f77258250edfc9d774de37b3a6918b9d2b8b5859
|
| 3 |
+
size 55948339
|
weights/added_IVF746_Flat_nprobe_1_Chamber_v2.index
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:61a7f6f9320e76e689d84e6739fb43045e687fb7e6aacc12f4587f779ecb1568
|
| 3 |
+
size 91922739
|
weights/kobov2.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ebf2826393f278168bbe6c9e6da614d75a6eeda408c33219dedf94688ed4f49c
|
| 3 |
+
size 57575716
|
weights/zeta.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:98797ebdcba9b7e6313e0e1c98c8f040b657d733079d09732ce035dca087990d
|
| 3 |
+
size 54996174
|