Update main.py

main.py

@@ -3,6 +3,7 @@ from fastapi.responses import JSONResponse, HTMLResponse
 from fastapi.templating import Jinja2Templates
 import joblib
 import pandas as pd
+import numpy as np
 from pydantic import BaseModel
 from sklearn.pipeline import Pipeline
 from sklearn.feature_selection import SelectKBest
@@ -14,6 +15,7 @@ from sklearn.feature_selection import f_classif
 from sklearn.impute import SimpleImputer 
 
 
+
 app = FastAPI()
 templates = Jinja2Templates(directory="templates")