3v324v23 commited on
Commit
0e30fdd
·
1 Parent(s): 16031b5
Files changed (1) hide show
  1. pages/Model_Evaluation.py +5 -1
pages/Model_Evaluation.py CHANGED
@@ -96,7 +96,11 @@ val_transform = transforms.Compose([
96
  # ---- Load Data from Hugging Face (cached) ----
97
  @st.cache_resource
98
  def load_test_data_from_huggingface():
99
- dataset = load_dataset("Ci-Dave/DDR_dataset_train_test/splits", data_files="test_labels.csv", split="test")
 
 
 
 
100
  df = dataset.to_pandas()
101
  csv_path = "test_labels_temp.csv"
102
  df.to_csv(csv_path, index=False)
 
96
  # ---- Load Data from Hugging Face (cached) ----
97
  @st.cache_resource
98
  def load_test_data_from_huggingface():
99
+ dataset = load_dataset(
100
+ "Ci-Dave/DDR_dataset_train_test",
101
+ data_files={"test": "splits/test_labels.csv"},
102
+ split="test"
103
+ )
104
  df = dataset.to_pandas()
105
  csv_path = "test_labels_temp.csv"
106
  df.to_csv(csv_path, index=False)