Update main.py

main.py

@@ -53,7 +53,7 @@ async def predict(input: InputFeatures):
     ])
 
     df = pd.DataFrame([input])
-    final_input = np.array(predict_input.fit_transform(df), dtype=np.str)  # Check predict_input, maybe it should be XGB
+    final_input = np.array(full_pipeline.fit_transform(df), dtype=np.str)  # Check predict_input, maybe it should be XGB
     prediction = model_input.predict(np.array([final_input]).reshape(1, -1))
 
     return prediction