Upload color_model.py with huggingface_hub

color_model.py

@@ -69,7 +69,9 @@ class ColorDataset(Dataset):
             Tuple of (image_tensor, token_tensor)
         """
         row = self.df.iloc[idx]
-        img = Image.open(config.column_local_image_path).convert("RGB")
+        # Fix: Get the image path from the row, not the column name
+        img_path = row[config.column_local_image_path]
+        img = Image.open(img_path).convert("RGB")
         img = self.transform(img)
         tokens = torch.tensor(self.tokenizer(row[config.text_column]), dtype=torch.long)
         return img, tokens