From 7ebcbdcab02182e93dd2d939aea0ff4646435749 Mon Sep 17 00:00:00 2001
From: Daniel Scott <6148219+nightvision04@users.noreply.github.com>
Date: Wed, 14 Dec 2022 21:41:59 -0700
Subject: [PATCH] ImageDataLoaders path fix

Module now found in fastai.vision.data
---
 initialise.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/initialise.py b/initialise.py
index fd3ce6e..5fde185 100644
--- a/initialise.py
+++ b/initialise.py
@@ -1,4 +1,5 @@
 from fastai.vision import *
+from fastai.vision.data import ImageDataLoaders
 from pathlib import Path
 
 # Data Augmentations
@@ -30,7 +31,7 @@ def get_tfms(): return get_transforms(do_flip = True,
 
 def get_model_data(path):
     path = Path(path)
-    data = ImageDataBunch.from_folder(path, 'train', 'valid', size = (375, 666), ds_tfms = get_tfms(), bs=1,
+    data = ImageDataLoaders.from_folder(path, 'train', 'valid', size = (375, 666), ds_tfms = get_tfms(), bs=1,
                                       resize_method = ResizeMethod.SQUISH,
                                       num_workers = 0
                                      ).normalize(imagenet_stats)