modified

pages/Model_Evaluation.py

@@ -102,13 +102,16 @@ def load_test_data_from_huggingface():
         split="test"
     )
     df = dataset.to_pandas()
-    
-    # 🔧 Add correct relative path to image files
-    df["new_path"] = "test/" + df["image"].astype(str)
+
+    # 🔍 Check actual column names if needed:
+    # st.write(df.columns)
+
+    # 🔧 Use correct column name for image paths
+    df["new_path"] = "test/" + df["filename"].astype(str)
 
     csv_path = "test_labels_temp.csv"
     df.to_csv(csv_path, index=False)
-    
+
     dataset = DDRDataset(csv_path=csv_path, transform=val_transform)
     return DataLoader(dataset, batch_size=32, shuffle=False)