Upload config.py with huggingface_hub

config.py

@@ -0,0 +1,12 @@
+import torch
+
+local_dataset_path = "data/data_with_local_paths.csv"
+color_model_path = "models/color_model.pt"
+hierarchy_model_path = "models/hierarchy_model.pth"
+main_model_path = "models/laion_explicable_model.pth"
+
+column_local_image_path = 'local_image_path'
+hierarchy_emb_dim = 64
+color_emb_dim = 16
+
+device = torch.device("mps")