Update utils.py

utils.py

@@ -2,13 +2,22 @@ import pandas as pd
 import re
 import ast
 import os
+from huggingface_hub import hf_hub_download
 
-# Get the directory where this utils.py file is located
+# Configuration
+REPO_ID = "theodoredc/MusicChordTransformer"
+CSV_FILENAME = "chords_mapping.csv"
+
+# Try to determine if we're running from a local repo or need to download
 current_dir = os.path.dirname(os.path.abspath(__file__))
-csv_path = os.path.join(current_dir, "chords_mapping.csv")
+local_csv_path = os.path.join(current_dir, CSV_FILENAME)
 
-# Read CSV into DataFrame
-chord_relations = pd.read_csv(csv_path)
+if os.path.exists(local_csv_path):
+    # Running locally (development)
+    csv_path = local_csv_path
+else:
+    # Download from HuggingFace Hub
+    csv_path = hf_hub_download(repo_id=REPO_ID, filename=CSV_FILENAME)
 
 # Create a dictionary with keys the "chords" and values the "degrees"
 chord_degrees = dict(zip(chord_relations['Chords'], chord_relations['Degrees']))