Spaces:
Paused
Paused
Commit
·
6f601b3
1
Parent(s):
1519720
Update main.py
Browse files
main.py
CHANGED
|
@@ -8,7 +8,9 @@ from accelerator import Accelerator
|
|
| 8 |
from answerer import Answerer
|
| 9 |
from mapper import Mapper
|
| 10 |
|
| 11 |
-
|
|
|
|
|
|
|
| 12 |
answerer = Answerer(
|
| 13 |
model="RWKV-5-World-3B-v2-20231118-ctx16k",
|
| 14 |
vocab="rwkv_vocab_v20230424",
|
|
|
|
| 8 |
from answerer import Answerer
|
| 9 |
from mapper import Mapper
|
| 10 |
|
| 11 |
+
try: mapper = Mapper("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
|
| 12 |
+
except: print("ERROR! cannot load Mapper model!")
|
| 13 |
+
|
| 14 |
answerer = Answerer(
|
| 15 |
model="RWKV-5-World-3B-v2-20231118-ctx16k",
|
| 16 |
vocab="rwkv_vocab_v20230424",
|