App.py

@@ -21,7 +21,7 @@ if bfs3_flag or Astar or Astar2:
    Pole.stoneFlag = True
 TreeFlag=False
 nnFlag=True
-newModel=False
+newModel=True
 
 pygame.init()
 show_console=True

neuralnetwork.py

@@ -9,7 +9,7 @@ from PIL import Image
 import random
 
 imageSize = (128, 128)
-labels = ['carrot', 'potato', 'tomato'] # musi być w kolejności alfabetycznej
+labels = ['carrot','corn', 'potato', 'tomato'] # musi być w kolejności alfabetycznej
 fertilizer = {labels[0]: 'kompost', labels[1]: 'saletra amonowa', labels[2]: 'superfosfat'}
 
 torch.manual_seed(42)