fix

app.py

@@ -15,7 +15,8 @@ def get_data_root():
         print("Downloading dataset from Hugging Face...")
         DATA_ROOT = snapshot_download(
             repo_id=HF_DATA_REPO,
-            repo_type="dataset"
+            repo_type="dataset",
+            token=os.environ["HF_TOKEN"]
         )
         print("Dataset downloaded to:", DATA_ROOT)
     return DATA_ROOT

upload_dataset_to_hf.py

@@ -1,15 +1,20 @@
 from huggingface_hub import HfApi, upload_folder
+import os
 
 HF_REPO = "doosaganesh/virtual-tryon-data"
 
-FOLDERS = [
+UPLOAD_FOLDERS = [
     "datasets",
-    "generated_tryon"
+    "generated_tryon",
+    "source_models"
 ]
 
+TEMP_ROOT = "hf_upload_temp"
+
 api = HfApi()
 
 print("Creating dataset repo if needed...")
+
 api.create_repo(
     repo_id=HF_REPO,
     repo_type="dataset",
@@ -17,15 +22,35 @@ api.create_repo(
     exist_ok=True
 )
 
-for folder in FOLDERS:
-    print(f"Uploading {folder}...")
+# Create temporary upload directory
+if os.path.exists(TEMP_ROOT):
+    import shutil
+    shutil.rmtree(TEMP_ROOT)
+
+os.makedirs(TEMP_ROOT, exist_ok=True)
+
+# Copy folders into temporary structure
+import shutil
+
+for folder in UPLOAD_FOLDERS:
+
+    if not os.path.exists(folder):
+        print(f"Skipping {folder} (not found)")
+        continue
+
+    dest = os.path.join(TEMP_ROOT, folder)
 
-    upload_folder(
-        repo_id=HF_REPO,
-        repo_type="dataset",
-        folder_path=folder,
-        path_in_repo=folder,
-        commit_message=f"Upload {folder}"
-    )
+    print(f"Preparing {folder}...")
+
+    shutil.copytree(folder, dest)
+
+print("Uploading dataset to Hugging Face...")
+
+upload_folder(
+    repo_id=HF_REPO,
+    repo_type="dataset",
+    folder_path=TEMP_ROOT,
+    commit_message="Upload datasets + generated_tryon + source_models"
+)
 
 print("Upload complete!")

upload_source_models.py

@@ -0,0 +1,9 @@
+from huggingface_hub import upload_folder
+
+upload_folder(
+    repo_id="doosaganesh/virtual-tryon-data",
+    repo_type="dataset",
+    folder_path="source_models",
+    path_in_repo="source_models",
+    commit_message="Upload source models"
+)