Update app.py

app.py

@@ -105,7 +105,7 @@ class DemandForecasting:
         # Return None if no date column is found
         return None
 
-    def preprocess_data(self, df_data: pd.DataFrame, list_columns) -> pd.DataFrame:
+    def preprocess_data(self, df_data: pd.DataFrame, list_columns, target_column) -> pd.DataFrame:
         """
         Preprocess the input DataFrame.
 
@@ -366,10 +366,10 @@ class DemandForecasting:
             input_data = self.parse_text(text, column_list)
 
             #load data
-            train_data = self.load_data(train_csv_path.name)
+            train_data = self.load_data(train_csv_path)
 
             #preprocess the train data
-            train_df = self.preprocess_data(train_data, column_list)
+            train_df = self.preprocess_data(train_data, column_list, target_column)
 
             # Train model and get validation predictions
             trained_model, y_val, y_val_pred, validation = self.train_model(train_df, target_column, column_list)