Update main.py

main.py

@@ -2,7 +2,12 @@ from fastapi import FastAPI, WebSocket
 from fastapi.responses import HTMLResponse
 
 from answerer import Answerer
+from mapper import Mapper
 
+mapper = Mapper(
+   repo="sentence-transformers",
+   model="multi-qa-distilbert-cos-v1",
+)
 answerer = Answerer(
   model="RWKV-5-World-3B-v2-20231118-ctx16k.pth",
   vocab="rwkv_vocab_v20230424",
@@ -52,16 +57,31 @@ HTML = """
 """
 
 @app.get("/")
-async def index():
-    return HTMLResponse(HTML)
+def index():
+  return HTMLResponse(HTML)
+
+@app.get("/map")
+def map():
+  return HTMLResponse(HTML)
 
 @app.websocket("/answer")
 async def answer(ws: WebSocket):
   await ws.accept()
+
+  print("ws accepted!")
       
   input = await ws.receive_text()
+
+  print("input received!")
+
+  await ws.send_text("OK!")
+
   output = answerer(input, 32)
+
+  print("output created!")
+
   for el in output:
-     await ws.send_text(el)
+    print(f"sent: '{el}'")
+    await ws.send_text(el)
 
   await ws.close()