update app

app.py

@@ -8,6 +8,18 @@ from sklearn.metrics import average_precision_score
 import matplotlib.pyplot as plt
 import torch
 import spaces
+import os
+from huggingface_hub import HfApi
+
+# Load Hugging Face token from the environment variable
+HF_TOKEN = os.getenv("HF_TOKEN")
+if HF_TOKEN is None:
+    raise ValueError("HF_TOKEN environment variable is not set. Please set it before running the script.")
+
+# Authenticate with Hugging Face
+api = HfApi()
+api.set_access_token(HF_TOKEN)
+
 
 # Check for GPU support and configure appropriately
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -34,8 +46,8 @@ def ndcg_at_k(relevance_labels, scores, k=10):
 
 # Load datasets
 datasets = {
-    "Relevance_Labels_Dataset": load_dataset("NAMAA-Space/Ar-Reranking-Eval")["train"],
-    "Positive_Negatives_Dataset": load_dataset("NAMAA-Space/Arabic-Reranking-Triplet-5-Eval")["train"]
+    "Relevance_Labels_Dataset": load_dataset("Omartificial-Intelligence-Space/re-ar-test-triplet" , token =HF_TOKEN )["train"],
+    "Positive_Negatives_Dataset": load_dataset("Omartificial-Intelligence-Space/re-ar-query-candidate" , token =HF_TOKEN )["train"]
 }
 
 @spaces.GPU