HinaCortus commited on
Commit
6c2e3e1
·
1 Parent(s): 6d205ac

Delete main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -26
main.py DELETED
@@ -1,26 +0,0 @@
1
- from fastapi import FastAPI
2
- from pydantic import BaseModel
3
- from app.model.model import predict_pipeline
4
- from app.model.model import __version__ as model_version
5
-
6
-
7
- app = FastAPI()
8
-
9
-
10
- class TextIn(BaseModel):
11
- text: str
12
-
13
-
14
- class PredictionOut(BaseModel):
15
- language: str
16
-
17
-
18
- @app.get("/")
19
- def home():
20
- return {"health_check": "OK", "model_version": model_version}
21
-
22
-
23
- @app.post("/predict", response_model=PredictionOut)
24
- def predict(payload: TextIn):
25
- language = predict_pipeline(payload.text)
26
- return {"language": language}