Fix weights generation

1 files changed, 3 insertions, 3 deletions

diff --git a/main.py b/main.py
index 9ca0a13..a44fbcb 100644
--- a/main.py
+++ b/main.py
@@ -1,12 +1,12 @@
-from PIL import Image
-from dataset_utils import DatasetUtils
+from dataset import Dataset
 from neural_network import NeuralNetwork
 
 LEARNING_RATE = .5
 INPUT_RESOLUTION = 278
 
+
 def main():
-    data = DatasetUtils()
+    data = Dataset()
     image = data.get_image('А/5a2f3c19c27bb.png')
     neural_network = NeuralNetwork(LEARNING_RATE, INPUT_RESOLUTION)