diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py
index 598c67a..f887078 100644
--- a/ml_pytorch_mlflow.py
+++ b/ml_pytorch_mlflow.py
@@ -136,7 +136,7 @@ def my_main(epochs):
         input_, target = val_ds[i]
         expected.append(float(target))
         predicted.append(float(prediction(input_, model)))
-        inputs.append(input_.numpy())
+        inputs = np.append(inputs, input_)
     
     inputs = pd.DataFrame(inputs, dtype=np.float64)
     inputs = inputs.to_numpy()