diff --git a/mlflow_model.py b/mlflow_model.py index 7d974ba..32c165f 100644 --- a/mlflow_model.py +++ b/mlflow_model.py @@ -57,7 +57,7 @@ if __name__ == "__main__": mlf.set_experiment("s434804") tracking_url_type_store = urlparse(mlf.get_tracking_uri()).scheme if tracking_url_type_store != "file": - mlf.keras.log_model(model, "country_vaccinations", registered_model_name="s434804", signature=signature, + mlf.keras.log_model(model, "country_vaccinations2", registered_model_name="s434804", signature=signature, input_example=input_example) else: mlf.keras.log_model(model, "vaccines_model", signature=signature, input_example=input_example)