DaniilAlpha commited on
Commit
479712a
·
1 Parent(s): 7155dc1

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -5
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",