From 091c794e5e7822fd573f52df435a1c6d595ce47f Mon Sep 17 00:00:00 2001 From: Wojciech Jarmosz Date: Sun, 23 May 2021 19:10:36 +0200 Subject: [PATCH] Clean prints --- mlflow_prediction.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/mlflow_prediction.py b/mlflow_prediction.py index 22abfc5..77f2f3b 100644 --- a/mlflow_prediction.py +++ b/mlflow_prediction.py @@ -16,7 +16,4 @@ model = mlflow.pytorch.load_model("my_model") with open(f'my_model/{input}', 'r') as file: json_data = json.load(file) -print(json_data) -print(json_data['inputs']) - print(model(torch.tensor(np.array(json_data['inputs'])).float()))