Update main.py

main.py

@@ -11,7 +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 
+from sklearn.impute import SimpleImputer 
 
 
 app = FastAPI()