Spaces:
Paused
Paused
Commit
·
479712a
1
Parent(s):
7155dc1
Update main.py
Browse files
main.py
CHANGED
@@ -1,4 +1,3 @@
|
|
1 |
-
from asyncio import sleep
|
2 |
from typing import Union
|
3 |
from fastapi import FastAPI
|
4 |
from fastapi.encoders import jsonable_encoder
|
@@ -9,10 +8,7 @@ from accelerator import Accelerator
|
|
9 |
from answerer import Answerer
|
10 |
from mapper import Mapper
|
11 |
|
12 |
-
mapper = Mapper(
|
13 |
-
repo="sentence-transformers",
|
14 |
-
model="multi-qa-distilbert-cos-v1",
|
15 |
-
)
|
16 |
answerer = Answerer(
|
17 |
model="RWKV-5-World-3B-v2-20231118-ctx16k.pth",
|
18 |
vocab="rwkv_vocab_v20230424",
|
|
|
|
|
1 |
from typing import Union
|
2 |
from fastapi import FastAPI
|
3 |
from fastapi.encoders import jsonable_encoder
|
|
|
8 |
from answerer import Answerer
|
9 |
from mapper import Mapper
|
10 |
|
11 |
+
mapper = Mapper("multi-qa-distilbert-cos-v1.bin")
|
|
|
|
|
|
|
12 |
answerer = Answerer(
|
13 |
model="RWKV-5-World-3B-v2-20231118-ctx16k.pth",
|
14 |
vocab="rwkv_vocab_v20230424",
|