Update prepare.py

prepare.py

@@ -1,30 +1,30 @@
-from huggingface_hub import hf_hub_download
-import shutil
-import os
-
-# Hugging Face model details
-REPO_ID = "xcurv/kcv-vanguard-day3-cnn-card-classification"
-MODEL_FILENAME = "model.h5"
-LABEL_JSON = "class.json"
-
-def download_file(repo_id, filename, dest_filename):
-    file_path = hf_hub_download(repo_id=repo_id, filename=filename)
-
-    # Get the absolute path of the actual file (resolves symlinks)
-    real_file_path = os.path.realpath(file_path)
-
-    # Ensure the resolved file exists before copying
-    if not os.path.exists(real_file_path):
-        raise FileNotFoundError(f"Resolved file path does not exist: {real_file_path}")
-
-    # Copy the actual file to the destination
-    shutil.copy2(real_file_path, dest_filename)
-    os.remove(real_file_path)
-
-# Download model if not present
-if not os.path.exists(MODEL_FILENAME):
-    download_file(REPO_ID, MODEL_FILENAME, MODEL_FILENAME)
-
-# Download label JSON if not present
-if not os.path.exists(LABEL_JSON):
-    download_file(REPO_ID, LABEL_JSON, LABEL_JSON)
+from huggingface_hub import hf_hub_download
+import shutil
+import os
+
+# Hugging Face model details
+REPO_ID = "riciii7/KCVanguard"
+MODEL_FILENAME = "model.h5"
+LABEL_JSON = "class.json"
+
+def download_file(repo_id, filename, dest_filename):
+    file_path = hf_hub_download(repo_id=repo_id, filename=filename)
+
+    # Get the absolute path of the actual file (resolves symlinks)
+    real_file_path = os.path.realpath(file_path)
+
+    # Ensure the resolved file exists before copying
+    if not os.path.exists(real_file_path):
+        raise FileNotFoundError(f"Resolved file path does not exist: {real_file_path}")
+
+    # Copy the actual file to the destination
+    shutil.copy2(real_file_path, dest_filename)
+    os.remove(real_file_path)
+
+# Download model if not present
+if not os.path.exists(MODEL_FILENAME):
+    download_file(REPO_ID, MODEL_FILENAME, MODEL_FILENAME)
+
+# Download label JSON if not present
+if not os.path.exists(LABEL_JSON):
+    download_file(REPO_ID, LABEL_JSON, LABEL_JSON)