Upload folder using huggingface_hub

app.py

@@ -8,9 +8,11 @@ import sys
 import custom_transformers
 
 # Patch sys.modules to let pickle find your custom classes
+sys.modules['__main__'] = sys.modules[__name__]
 sys.modules['__main__'].ProductPrefixExtractor = custom_transformers.ProductPrefixExtractor
 sys.modules['__main__'].StoreAgeBinner = custom_transformers.StoreAgeBinner
 sys.modules['__main__'].store_age_transformer = custom_transformers.store_age_transformer
+sys.modules['__main__'].convert_year_to_age = custom_transformers.convert_year_to_age
 
 # Initialize Flask app
 sales_predictor_api = Flask("Retail Sales Prediction API")