Update utils.py

utils.py

@@ -9,14 +9,15 @@ from huggingface_hub import Repository, HfApi
 import os
 tqdm.pandas()
 
-# Authentifizierung (stellen Sie sicher, dass Ihr Hugging Face Token gesetzt ist)
-api = HfApi(
-    endpoint="https://huggingface.co/spaces/Add1E/SpotifyHitPrediction",
-    token=os.getenv("token")
-)
+api = HfApi()
+token = HfFolder.get_token()  # Das Token wird aus den Hugging Face Secrets abgerufen
 
-# Erstellen oder Klonen Sie ein Repository (wenn noch nicht vorhanden)
-repo = Repository(local_dir="https://huggingface.co/spaces/Add1E/SpotifyHitPrediction", clone_from="Add1E/SpotifyHitPrediction")
+# Überprüfen, ob das Token vorhanden ist
+if token is None:
+    raise ValueError("Hugging Face API-Token ist nicht gesetzt.")
+
+# Klonen Sie das Repository (dies wird in Ihrem Space ausgeführt)
+repo = Repository(local_dir="SpotifyHitPrediction", clone_from="Add1E/SpotifyHitPrediction", use_auth_token=token)
 
 def predict_popularity(features, trainset):
     predictions = [None] * 2
@@ -33,7 +34,7 @@ def addToCsvAndTrain(trainset):
     ]
     neues_df = pd.DataFrame(trainset, columns= data.columns)
     df = pd.concat([data, neues_df], ignore_index=True)
-    df.to_csv('SpotifyHitPrediction/top50.csv', index=False)
+    df.to_csv('top50.csv', index=False)
     repo.git_add('top50.csv')
     repo.git_commit("Add top50.csv")
     repo.git_push()