From 49db999cc957892411ba0182a8e8dd76786c413e Mon Sep 17 00:00:00 2001
From: happy531 <jedrzejwesolowski177@gmail.com>
Date: Thu, 26 May 2022 20:55:34 +0200
Subject: [PATCH] repair getPrediction

---
 NeuralNetwork/prediction.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/NeuralNetwork/prediction.py b/NeuralNetwork/prediction.py
index d6e27ee..6ee634a 100644
--- a/NeuralNetwork/prediction.py
+++ b/NeuralNetwork/prediction.py
@@ -5,9 +5,9 @@ from PIL import Image
 from NeuralNetwork import NeuralNetwork
 
 def getPrediction(img_path):
-
+    
     # Inicjacja sieci neuronowej
-    neural_net = NeuralNetwork()
+    neural_net = NeuralNetwork.NeuralNetwork()
     PATH = './trained_nn.pth'
     img = Image.open(img_path)
     transform_tensor = transforms.ToTensor()(img).unsqueeze_(0)