BookSumBeta

Sleeping

npc0 commited on Aug 20, 2023

Commit

308c6df

1 Parent(s): 4f75c30

Update api.py

Files changed (1) hide show

api.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import os
 from typing import Union
-from fastapi import FastAPI
 from epub2txt import epub2txt
 from fastllm_pytools import llm
@@ -12,7 +12,7 @@ prompt = os.getenv("prompt")
 app = FastAPI()
-@app.websockets("/{f_name}/ws")
 async def read_root(*,
                     websocket: WebSocket,
                    f_name: str):

 import os
 from typing import Union
+from fastapi import FastAPI, WebSocket
 from epub2txt import epub2txt
 from fastllm_pytools import llm
 app = FastAPI()
+@app.websocket("/{f_name}/ws")
 async def read_root(*,
                     websocket: WebSocket,
                    f_name: str):