From 4edda8a8047c81aa4c86a54611d11be30acfdb52 Mon Sep 17 00:00:00 2001 From: Kamila Date: Sun, 15 May 2022 12:55:35 +0200 Subject: [PATCH] mlflow attempt task 2 prediction --- predict.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/predict.py b/predict.py index f8cdb35..af22b09 100644 --- a/predict.py +++ b/predict.py @@ -3,14 +3,13 @@ import mlflow import numpy as np import sys import tarfile -import os + file = tarfile.open('mlruns.tar.gz') file.extractall('./ml') input = str((sys.argv[1:])[0]) PATH = "ml/mlruns/1/f65f936936024133a2c03e1e486ba9cf/artifacts/model/" -print(os.listdir(PATH)) -model = mlflow.pytorch.load_model(f"{PATH}/MLmodel") +model =mlflow.pyfunc.load_model(f"{PATH}/MLmodel") with open(f'[PATH]/{input}', 'r') as file: json_data = json.load(file)