repair getPrediction

2022-05-26 20:55:34 +02:00 · 2022-05-26 20:55:34 +02:00 · 49db999cc9
commit 49db999cc9
parent 134e45de18
1 changed files with 2 additions and 2 deletions
--- 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)