Spaces:
Paused
Paused
Commit
·
589d913
1
Parent(s):
dff2e80
Fixed bug.
Browse files
main.py
CHANGED
@@ -10,7 +10,7 @@ from answerer import Answerer
|
|
10 |
from mapper import Mapper
|
11 |
|
12 |
try: mapper = Mapper("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
|
13 |
-
except: print("ERROR! cannot load Mapper model
|
14 |
|
15 |
answerer = Answerer(
|
16 |
model="RWKV-5-World-3B-v2-20231118-ctx16k",
|
@@ -69,7 +69,8 @@ def index():
|
|
69 |
@app.websocket("/accelerate")
|
70 |
async def answer(ws: WebSocket):
|
71 |
await accelerator.connect(ws)
|
72 |
-
while accelerator.connected():
|
|
|
73 |
|
74 |
@app.post("/map")
|
75 |
def map(query: Union[str, None], items: Union[list[str], None]):
|
|
|
10 |
from mapper import Mapper
|
11 |
|
12 |
try: mapper = Mapper("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
|
13 |
+
except Exception as e: print(f"ERROR! cannot load Mapper model!\n{e}")
|
14 |
|
15 |
answerer = Answerer(
|
16 |
model="RWKV-5-World-3B-v2-20231118-ctx16k",
|
|
|
69 |
@app.websocket("/accelerate")
|
70 |
async def answer(ws: WebSocket):
|
71 |
await accelerator.connect(ws)
|
72 |
+
while accelerator.connected():
|
73 |
+
await sleep(10)
|
74 |
|
75 |
@app.post("/map")
|
76 |
def map(query: Union[str, None], items: Union[list[str], None]):
|