Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import os
|
2 |
from typing import Union
|
3 |
|
4 |
-
from fastapi import FastAPI
|
5 |
from epub2txt import epub2txt
|
6 |
|
7 |
from fastllm_pytools import llm
|
@@ -12,7 +12,7 @@ prompt = os.getenv("prompt")
|
|
12 |
app = FastAPI()
|
13 |
|
14 |
|
15 |
-
@app.
|
16 |
async def read_root(*,
|
17 |
websocket: WebSocket,
|
18 |
f_name: str):
|
|
|
1 |
import os
|
2 |
from typing import Union
|
3 |
|
4 |
+
from fastapi import FastAPI, WebSocket
|
5 |
from epub2txt import epub2txt
|
6 |
|
7 |
from fastllm_pytools import llm
|
|
|
12 |
app = FastAPI()
|
13 |
|
14 |
|
15 |
+
@app.websocket("/{f_name}/ws")
|
16 |
async def read_root(*,
|
17 |
websocket: WebSocket,
|
18 |
f_name: str):
|