update path

scripts/app.py

@@ -32,8 +32,10 @@ def load_artifacts():
     
     # Configuration
     CHECKPOINT_PATH = "checkpoints/sasrec-epoch=06-val_hitrate@10=0.3629.ckpt"
-    DATA_FOLDER = "data/"
     DATA_REPO_ID = "Deathshot78/RetailRocket-Recommender-Data"
+    # FIX: Define the correct, full path to where the data will be after download.
+    # The nested structure comes from the path within the HF Hub dataset repo.
+    DATA_FOLDER = "data/RetailRocket-Recommender-Data/data"
 
     # --- Download Data from Hugging Face Hub ---
     print(f"Downloading data from Hugging Face Hub repo: {DATA_REPO_ID}")
@@ -47,9 +49,9 @@ def load_artifacts():
     for filename in files_to_download:
         hf_hub_download(
             repo_id=DATA_REPO_ID,
-            # FIX: Update the path to match the nested directory structure on the Hub
+            # The path to the file within the dataset repo
             filename=f"data/RetailRocket-Recommender-Data/data/{filename}",
-            local_dir=".",              # Download to the root of the Space
+            local_dir=".",              # Download to the root of the Space, preserving structure
             repo_type="dataset"
         )
     print("All data files downloaded successfully.")
@@ -64,12 +66,14 @@ def load_artifacts():
     MODEL.eval()
 
     print("Preparing data from downloaded files...")
+    # Use the corrected DATA_FOLDER path
     train_set, validation_set, test_set = prepare_data(data_folder=DATA_FOLDER)
     
     DATAMODULE = SASRecDataModule(train_set, validation_set, test_set)
     DATAMODULE.setup()
     
     print("Loading item and category maps...")
+    # Use the corrected DATA_FOLDER path
     ITEM_CATEGORY_MAP = load_item_properties(data_folder=DATA_FOLDER)
     CATEGORY_PARENT_MAP = load_category_tree(data_folder=DATA_FOLDER)
     
@@ -184,4 +188,4 @@ if __name__ == "__main__":
     
     # For local testing, this creates a shareable link.
     # On Hugging Face Spaces, this is not strictly necessary but doesn't hurt.
-    iface.launch(share=True)
+    iface.launch(share=True)