From 99aa9a85a0fc9561d51d9d64d323e9e9f6fe5f73 Mon Sep 17 00:00:00 2001 From: Wojciech Jarmosz Date: Sun, 23 May 2021 18:28:44 +0200 Subject: [PATCH] Fix prediction script --- mlflow_prediction.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlflow_prediction.py b/mlflow_prediction.py index 32c7efb..75791a2 100644 --- a/mlflow_prediction.py +++ b/mlflow_prediction.py @@ -16,4 +16,4 @@ model = mlflow.pytorch.load_model("my_model") with open(f'my_model/{input}', 'r') as file: json_data = json.load(file)['inputs'] -print(model(torch.tensor(np.array(json_data['inputs'])).float())) \ No newline at end of file +print(model(torch.tensor(np.array(json_data['inputs'])).float()))