Update src/streamlit_app.py

src/streamlit_app.py

@@ -46,13 +46,13 @@ input_df = pd.DataFrame(input_data)
 
 # Custom transformer to replace 'reg' with 'Regular' in Product_Sugar_Content
 class SugarContentReplacer(BaseEstimator, TransformerMixin):
-    def fit(self, X, y=None):
+    def fit(self, input_df, y=None):
         return self
 
-    def transform(self, X):
-        X = X.copy()
-        X['Product_Sugar_Content'] = X['Product_Sugar_Content'].replace('reg', 'Regular')
-        return X
+    def transform(self, input_df):
+        input_df = input_df.copy()
+        input_df['Product_Sugar_Content'] = input_df['Product_Sugar_Content'].replace('reg', 'Regular')
+        return input_df
 
     # Add get_feature_names_out method
     def get_feature_names_out(self, input_features=None):