| 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() | |