From 77a5e3c22ae60e54df14cd5446f5470d3e557da5 Mon Sep 17 00:00:00 2001 From: Filip Izydorczyk Date: Tue, 8 Jun 2021 18:10:26 +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 fdeb1f7..02dc1f0 100644 --- a/predict/predict.py +++ b/predict/predict.py @@ -3,7 +3,7 @@ import mlflow import pandas as pd -model = mlflow.sklearn.load_model("./prediction_data/movies_imdb") +model = mlflow.keras.load_model("./prediction_data/movies_imdb") with open('./prediction_data/movies_imdb/input_example.json', 'r') as datafile: