Update main.py

main.py

@@ -29,7 +29,6 @@ HTML = """
     const prompt = document.getElementById("prompt");
     const output = document.getElementById("output");
 
-    // TODO!
     const ws = new WebSocket("wss://daniilalpha-answerer-api.hf.space/answer"); 
     ws.onmessage = (e) => answer(e.data);
 
@@ -62,23 +61,7 @@ async def answer(ws: WebSocket):
 
   input = await ws.receive_text()
   output = answerer(input, 32)
-  async for el in output:
-    await ws.send_text(el) 
-
-  await ws.close()
-
-async def answerer_test():
-  for i in range(10000):
-    yield f"i"
-
-@app.websocket("/answer_test")
-async def answer(ws: WebSocket):
-  await ws.accept()
-
-  input = await ws.receive_text()
-  output = answerer_test()
   for el in output:
-    print(el)
     await ws.send_text(el) 
 
   await ws.close()