Skip to content

Commit cb94374

Browse files
committed
refactor: model import to use module reference
1 parent 4193640 commit cb94374

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

app/server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
from starlette.responses import PlainTextResponse, Response
77
from starlette.routing import Route, Router
88

9+
from . import model
910
from .metrics import *
10-
from .model import async_predict as model_predict
1111

1212

1313
async def predict(request: Request):
@@ -17,7 +17,7 @@ async def predict(request: Request):
1717

1818
try:
1919
text = request.query_params.get("text")
20-
result = await model_predict(text) if text else False
20+
result = await model.async_predict(text) if text else False
2121

2222
label = "toxic" if result else "non_toxic"
2323
REQUEST_COUNT.labels(endpoint=endpoint, result=label).inc()

0 commit comments

Comments
 (0)