diff --git a/main.py b/main.py
index 2b6ba577430b3c592fd886909f080aea49254f9b..768e77a90b7b3989b0fb70ac553723d096469606 100644
--- a/main.py
+++ b/main.py
@@ -22,6 +22,7 @@ def create_inputs_and_outputs(folder_path,file_name):
     for idx in range(number_of_images):
         file_name = df['file_name'][idx]
         file_path = os.path.join(image_folder_dir,file_name)
+        print(file_path)
         original_image = cv2.imread(file_path,1)
         image_array[idx] = original_image/255.0
         label_array[idx] = df['number_of_leaves'][idx]
@@ -79,7 +80,7 @@ class Leaf_counting_model(tf.keras.Model):
                 monitor='val_loss',
                 patience=10,mode='min')
         history = self.model3.fit(train_dataset,
-                epochs=3,
+                epochs=100,
                 validation_data=val_dataset,
                 callbacks=[early_stopping])
         print(history.history)