Improving model, fix plate prediction

main.py

@@ -21,7 +21,9 @@ if __name__ == "__main__":
     goal = None
     path = ''
 
+    # Maksymilian
     go = 0
+    rand = 0
     while True:
 
         for event in pygame.event.get():
@@ -49,13 +51,12 @@ if __name__ == "__main__":
                 if event.key == pygame.K_m:
 
                     tabPos = [[1, 2], [1, 5], [1, 8], [5, 4], [5, 8], [8, 2], [8, 5], [8, 8], [12, 3], [12, 7]]
-                    rand = randrange(8)
 
                     if not [waiter.X, waiter.Y] in tabPos or go == 0:
                         if [waiter.X, waiter.Y] in tabPos:
                             model = 'waiter_' + waiter.direction
                             graphics.clear(waiter.X, waiter.Y - 1)
-                        # save_dataset()
+                        rand = randrange(9)
                         x = tabPos[rand][0]
                         y = tabPos[rand][1]