Update product_return_prediction/api.py

product_return_prediction/api.py

@@ -117,7 +117,7 @@ async def predict(products: ProductRequest):
     scaler_file: Path = MODELS_DIR / "scaler.pkl"
 
     hf_token = os.getenv("inventory_data")
-    inventory_data = load_dataset("molinari135/armani-inventory", token=hf_token, data_files="inventory.tsv")
+    dataset = load_dataset("molinari135/armani-inventory", token=hf_token, data_files="inventory.tsv")
     inventory = pd.DataFrame(dataset['train'])
 
     filtered_inventory = filter_inventory_by_combinations(