Spaces:

matthoffner
/

starchat-ggml

Paused

matthoffner commited on May 29, 2023

Commit

b0aa891

1 Parent(s): d8bf59e

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -42,11 +42,11 @@ async def demo():
     <html>
         <body>
             <div id="content"></div>
             <script>
               var source = new EventSource("https://matthoffner-starchat-alpha.hf.space/stream");
               source.onmessage = function(event) {
-                document.getElementById("content").innerHTML += event.data
               };
             </script>
@@ -56,7 +56,7 @@ async def demo():
     return HTMLResponse(content=html_content, status_code=200)
 @app.get("/stream")
-async def chat(prompt = "Write a simple express erver"):
     tokens = llm.tokenize(prompt)
     async def server_sent_events(chat_chunks, llm):
         yield prompt

     <html>
         <body>
             <div id="content"></div>
+            <script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
             <script>
               var source = new EventSource("https://matthoffner-starchat-alpha.hf.space/stream");
               source.onmessage = function(event) {
+                document.getElementById("content").innerHTML += marked.parse(event.data)
               };
             </script>
     return HTMLResponse(content=html_content, status_code=200)
 @app.get("/stream")
+async def chat(prompt = "Write a simple expres server"):
     tokens = llm.tokenize(prompt)
     async def server_sent_events(chat_chunks, llm):
         yield prompt