File size: 441 Bytes
4d6d610 9e290f8 4d6d610 9e290f8 4d6d610 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import joblib
from pydantic import BaseModel
from src.dup_ques.preprocess import get_x
# SCHEMA
class Schema(BaseModel):
ques1: str
ques2: str
# Request Handler
def dup_ques(req):
ques1 = req.ques1
ques2 = req.ques2
X = get_x(ques1, ques2)
y = predict(X)
return y
# PIPELINE
PIPELINE_PATH = "./src/dup_ques/pipeline.pkl"
def predict(X):
pipeline = joblib.load(PIPELINE_PATH)
return pipeline.predict_proba(X).round(3).tolist()
|