diff --git a/evaluate.py b/evaluate.py
index fcfb97d..4a3771c 100644
--- a/evaluate.py
+++ b/evaluate.py
@@ -2,6 +2,7 @@ import tensorflow as tf
 from tensorflow import keras
 from matplotlib import pyplot as plt
 from matplotlib.ticker import MaxNLocator
+import numpy as np
 
 # Załadowanie modelu z pliku
 model = keras.models.load_model('lego_reg_model')