Add logs

2021-05-23 18:57:18 +02:00 · 2021-05-23 18:57:18 +02:00 · c221c48a4a
commit c221c48a4a
parent 99aa9a85a0
1 changed files with 4 additions and 1 deletions
--- a/mlflow_prediction.py
+++ b/mlflow_prediction.py
@ -14,6 +14,9 @@ input = str(arguments[0])
 model = mlflow.pytorch.load_model("my_model")

 with open(f'my_model/{input}', 'r') as file:
-    json_data = json.load(file)['inputs']
+    json_data = json.load(file)
+
+print(json_data)
+print(json_data['inputs'])

 print(model(torch.tensor(np.array(json_data['inputs'])).float()))