diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py
index 5dbba56..ab1bf37 100644
--- a/ml_pytorch_mlflow.py
+++ b/ml_pytorch_mlflow.py
@@ -142,7 +142,8 @@ def my_main(epochs):
     
     # inputss, targetss = val_ds
     inputss = pd.DataFrame(inputss, dtype=np.float64)
-    inputss = inputss.to_numpy()
+    # inputss = inputss.to_numpy()
+    inputsss = inputss.values
 
     MSE = mean_squared_error(expected, predicted)
     MAE = mean_absolute_error(expected, predicted)
@@ -156,10 +157,10 @@ def my_main(epochs):
             input_, target = val_ds[i]
             file.write(str(predict_single(input_, target, model)))
 
-    print(inputss)
+    print(inputsss)
 
-    input_example = inputss[0]
-    signature = infer_signature(inputss, expected)
+    input_example = inputsss[0]
+    signature = infer_signature(inputsss, expected)
     tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
 
     if tracking_url_type_store != "file":