diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py
index ceb39f0..9d95372 100644
--- a/ml_pytorch_mlflow.py
+++ b/ml_pytorch_mlflow.py
@@ -137,10 +137,10 @@ def my_main(epochs):
         input_, target = val_ds[i]
         expected.append(float(target))
         predicted.append(float(prediction(input_, model)))
-        # inputss.append(input_)
+        inputss[i], targetss[i] = val_ds[i]
     
     
-    inputss, targetss = val_ds
+    # inputss, targetss = val_ds
     # inputss = pd.DataFrame(inputss, dtype=np.float64)
     # inputss = inputss.to_numpy()