s434704/ium_434704
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Change flavor of model prediction
Some checks failed
s434704-predict-s426206/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Wojciech Jarmosz 2021-05-23 17:04:25 +02:00
parent 31201b169b
commit 724679e783
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlflow_prediction.py
View File

@ -9,7 +9,7 @@ arguments = sys.argv[1:]
input = str(arguments[0])
model = mlflow.pytorchg.load_model("my_model")
model = mlflow.pytorch.load_model("my_model")
with open('my_model/{input}', 'r') as file:
json_data = json.load(file)['inputs']