Update main.py

main.py

@@ -11,6 +11,7 @@ from xgboost import XGBClassifier
 from sklearn.preprocessing import StandardScaler
 from sklearn.compose import ColumnTransformer
 from sklearn.feature_selection import f_classif
+from sklearn.preprocessing import SimpleImputer 
 
 
 app = FastAPI()