diff --git a/machine_learning/neuralModel.h5 b/machine_learning/neuralModel.h5
index d008213..aa65f50 100644
Binary files a/machine_learning/neuralModel.h5 and b/machine_learning/neuralModel.h5 differ
diff --git a/machine_learning/neuralNetwork.py b/machine_learning/neuralNetwork.py
index e533f30..145cafd 100644
--- a/machine_learning/neuralNetwork.py
+++ b/machine_learning/neuralNetwork.py
@@ -1,8 +1,11 @@
+import os
 from keras.preprocessing.image import ImageDataGenerator
 from keras.models import Sequential
 from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
 
 train_data_dir = "garbage_photos"
+location = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
+train_data_dir = os.path.join(location, train_data_dir)
 
 input_shape = (150, 150, 3)
 num_classes = 5