DaniilAlpha commited on
Commit
589d913
·
1 Parent(s): dff2e80

Fixed bug.

Browse files
Files changed (1) hide show
  1. main.py +3 -2
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(): sleep(10)
 
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]):