Fix prediction script

2021-05-23 18:28:44 +02:00 · 2021-05-23 18:28:44 +02:00 · 99aa9a85a0
commit 99aa9a85a0
parent 33c19f9112
1 changed files with 1 additions and 1 deletions
--- 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()))
+print(model(torch.tensor(np.array(json_data['inputs'])).float()))