Update product_return_prediction/api.py

product_return_prediction/api.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel, Field
 import pandas as pd
 import json
+import os
 import pickle
 from pathlib import Path
 from product_return_prediction.dataset import prepare_inventory, scale_data_with_trained_scaler
@@ -114,7 +115,9 @@ async def predict(products: ProductRequest):
     model_path: Path = MODELS_DIR / "svm.pkl"
     scaler_file: Path = MODELS_DIR / "scaler.pkl"
 
-    inventory = pd.read_csv(inventory_path, sep='\t')
+    hf_token = os.getenv("inventory_data")
+    inventory_data = load_dataset("molinari135/armani-inventory", token=hf_token, data_files="inventory.tsv")
+    inventory = pd.DataFrame(dataset['train'])
 
     filtered_inventory = filter_inventory_by_combinations(
         inventory, products.models, products.fabrics, products.colours