Upload evaluation/main_model_evaluation.py with huggingface_hub

evaluation/main_model_evaluation.py

@@ -19,7 +19,7 @@ import warnings
 warnings.filterwarnings('ignore')
 from transformers import CLIPProcessor, CLIPModel as CLIPModel_transformers
 
-from config import main_model_path, hierarchy_model_path, color_emb_dim, hierarchy_emb_dim, local_dataset_path, column_local_image_path
+from config import main_model_path, hierarchy_model_path, color_emb_dim, hierarchy_emb_dim, local_dataset_path, column_local_image_path, fashion_mnist_test_path
 
 
 def create_fashion_mnist_to_hierarchy_mapping(hierarchy_classes):
@@ -176,7 +176,7 @@ class FashionMNISTDataset(Dataset):
 
 def load_fashion_mnist_dataset(max_samples=1000, hierarchy_classes=None):
     print("📊 Loading Fashion-MNIST test dataset...")
-    df = pd.read_csv("/Users/leaattiasarfati/Desktop/docs/search/old/MainModel/data/fashion-mnist_test.csv")
+    df = pd.read_csv(fashion_mnist_test_path)
     print(f"✅ Fashion-MNIST dataset loaded: {len(df)} samples")
     
     # Create mapping if hierarchy classes are provided