Upload folder using huggingface_hub

app.py

@@ -4,7 +4,13 @@ import numpy as np
 import joblib
 import pandas as pd
 from flask import Flask, request, jsonify
-from custom_transformers import ProductPrefixExtractor, StoreAgeBinner, store_age_transformer, preprocessor
+import sys
+import custom_transformers
+
+# Patch sys.modules to let pickle find your custom classes
+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
 
 # Initialize Flask app
 sales_predictor_api = Flask("Retail Sales Prediction API")