From dc3d019d1ea5d909f46bf3c711086f57d9a3645d Mon Sep 17 00:00:00 2001 From: Filip Izydorczyk Date: Tue, 8 Jun 2021 18:01:11 +0200 Subject: [PATCH] predict anothre try --- predict/predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/predict/predict.py b/predict/predict.py index 696c8fb..0904f14 100644 --- a/predict/predict.py +++ b/predict/predict.py @@ -3,7 +3,7 @@ import mlflow import pandas as pd -model = mlflow.keras.load_model("./prediction_data/movies_imdb/MLmodel") +model = mlflow.sklearn.load_model("./prediction_data/movies_imdb/MLmodel") with open('./prediction_data/movies_imdb/input_example.json', 'r') as datafile: