Update main.py

main.py

@@ -32,9 +32,8 @@ HTML = """
 
     const ws = new WebSocket("wss://huggingface.co/spaces/DaniilAlpha/answerer-api:8000/answer");  
     ws.onopen = () => console.log("socket open!");
-    ws.onclose = () => console.log("socket closed!");
-    ws.onerror = (e) => console.error(`error: ${JSON.stringify(e)}`);
-    ws.onmessage = (event) => output.innerText = event.data;
+    ws.onclose = (e) => console.log(`socket closed with code ${e.code}!`);
+    ws.onmessage = (e) => output.innerText = e.data;
 
     function ask(event) {
       ws.send(prompt.value);