File size: 410 Bytes
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 |
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 = joblib.load("./src/dup_ques/pipeline.pkl")
def predict(X):
return pipeline.predict_proba(X).round(3).tolist()
|