fix: Change import root

slink-to-unlock · Jun 1, 2024 · 41f761f · 41f761f
1 parent 1b2ba3d
commit 41f761f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autosink_model_training/preprocess.py b/autosink_model_training/preprocess.py
@@ -1,5 +1,5 @@
 from transformers import AutoFeatureExtractor
-from dataset import load_custom_dataset
+from autosink_model_training.dataset import load_custom_dataset
 from torchvision.transforms import Compose, Resize, Normalize, ToTensor, RandomHorizontalFlip, ColorJitter, RandomApply, RandomVerticalFlip
 
 # Function to preprocess and augment images