Update main.py

main.py

@@ -43,8 +43,9 @@ async def predict(input: InputFeatures):
     #creating pipelines
     num_pipeline= Pipeline([('imputer', SimpleImputer()),('scaler', StandardScaler())])
 
-
+    
     full_pipeline=ColumnTransformer([('num_pipe',num_pipeline,num_attr)])
+    df = pd.DataFrame([input])
     full_pipeline.fit(df)  # Fit the full pipeline on the DataFrame
 
     final_input = np.array(full_pipeline.transform(df), dtype=np.str)
@@ -66,7 +67,9 @@ async def predict(input: InputFeatures):
 
     #     #final_input = np.array(full_pipeline.transform(df), dtype=np.str)
 
-    
+    #print(model_input)
+    #df = pd.DataFrame([input])
+    #final_input = np.array(predict_input.fit_transform(df), dtype = np.str)