one more time

2021-05-24 01:01:49 +02:00 · 2021-05-24 01:01:49 +02:00 · aa78c08876
commit aa78c08876
parent 315a3d3bda
1 changed files with 1 additions and 1 deletions
--- a/Zadanie_09_MLflow_Predict_Registry.py
+++ b/Zadanie_09_MLflow_Predict_Registry.py
@ -10,7 +10,7 @@ for mv in client.search_model_versions(f"name='{model_name}'"):
    if int(mv.version) > version:
        version = int(mv.version)

-model = mlflow.pytorch.load_model(
+model = mlflow.keras.load_model(
    model_uri=f"models:/{model_name}/{version}"
 )