Update main.py

main.py

@@ -5,7 +5,7 @@ from answerer import Answerer
 
 answerer = Answerer(
   repo="BlinkDL/rwkv-5-world", 
-  model="RWKV-5-World-3B-v2-20231118-ctx16k",
+  filename="RWKV-5-World-3B-v2-20231118-ctx16k",
   vocab="rwkv_vocab_v20230424",
   strategy="cpu bf16",
   ctx_limit=16*1024,
@@ -25,13 +25,15 @@ HTML = """
     <input type="submit" value="SEND" />
   </form>
 
-  <p id="output"></p>
+  <p id="output" placeholder="Output"></p>
   <script>
     const prompt = document.getElementById("prompt");
     const output = document.getElementById("output");
 
     const ws = new WebSocket("ws://huggingface.co/spaces/DaniilAlpha/answerer-api:8000/answer");  
-    ws.onmessage = function (event) => output.innerText = event.data;
+    ws.onmessage = function (event) {
+      output.innerText = event.data;
+    }
 
     function ask(event) {
       ws.send(prompt.value);
@@ -54,4 +56,4 @@ async def answer(ws: WebSocket):
   input = await ws.receive_text()
   stream = answerer(input, 256)
   for el in stream:
-    await ws.send_text(el)
+    await ws.send_text(el)