Update product_return_prediction/api.py

product_return_prediction/api.py

@@ -12,6 +12,10 @@ from product_return_prediction.monitoring import instrumentator
 from product_return_prediction.config import MODELS_DIR, EXTERNAL_DATA_DIR
 from huggingface_hub import hf_hub_download
 
+from huggingface_hub import login
+
+login(token=os.getenv("HUGGINGFACE_TOKEN"))
+
 app = FastAPI(
     title="Product Return Prediction API",
     description="This API predicts whether a product will be returned based on products and user behavior.",