Spaces:
Runtime error
Runtime error
Commit
·
a63c878
1
Parent(s):
69bd1d2
Delete app
Browse files- app/main.py +0 -26
- app/model/model.py +0 -40
- app/model/trained_pipeline-0.1.0.pkl +0 -3
app/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}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
app/model/model.py
DELETED
@@ -1,40 +0,0 @@
|
|
1 |
-
import pickle
|
2 |
-
import re
|
3 |
-
from pathlib import Path
|
4 |
-
|
5 |
-
__version__ = "0.1.0"
|
6 |
-
|
7 |
-
BASE_DIR = Path(__file__).resolve(strict=True).parent
|
8 |
-
|
9 |
-
|
10 |
-
with open(f"{BASE_DIR}/trained_pipeline-{__version__}.pkl", "rb") as f:
|
11 |
-
model = pickle.load(f)
|
12 |
-
|
13 |
-
|
14 |
-
classes = [
|
15 |
-
"Arabic",
|
16 |
-
"Danish",
|
17 |
-
"Dutch",
|
18 |
-
"English",
|
19 |
-
"French",
|
20 |
-
"German",
|
21 |
-
"Greek",
|
22 |
-
"Hindi",
|
23 |
-
"Italian",
|
24 |
-
"Kannada",
|
25 |
-
"Malayalam",
|
26 |
-
"Portugeese",
|
27 |
-
"Russian",
|
28 |
-
"Spanish",
|
29 |
-
"Sweedish",
|
30 |
-
"Tamil",
|
31 |
-
"Turkish",
|
32 |
-
]
|
33 |
-
|
34 |
-
|
35 |
-
def predict_pipeline(text):
|
36 |
-
text = re.sub(r'[!@#$(),\n"%^*?\:;~`0-9]', " ", text)
|
37 |
-
text = re.sub(r"[[]]", " ", text)
|
38 |
-
text = text.lower()
|
39 |
-
pred = model.predict([text])
|
40 |
-
return classes[pred[0]]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
app/model/trained_pipeline-0.1.0.pkl
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:ab7135c7421b2636cd9fe030d05b883c5ba19ccd13e57195e755a9e43df2bc53
|
3 |
-
size 10251835
|
|
|
|
|
|
|
|