Update main.py

main.py

@@ -45,10 +45,10 @@ async def predict(input: InputFeatures):
 
 
     full_pipeline=ColumnTransformer([('num_pipe',num_pipeline,num_attr)])
-        full_pipeline.fit(df)  # Fit the full pipeline on the DataFrame
+    full_pipeline.fit(df)  # Fit the full pipeline on the DataFrame
 
     final_input = np.array(full_pipeline.transform(df), dtype=np.str)
-        prediction = model_input.predict(final_input)
+    prediction = model_input.predict(final_input)
 
     XGB = Pipeline([
         ("col_trans", full_pipeline),  # You need to define full_pipeline