diff --git a/training_mlflow.py b/training_mlflow.py
index ea678e7..c783289 100644
--- a/training_mlflow.py
+++ b/training_mlflow.py
@@ -118,9 +118,9 @@ def my_main(epochs):
     tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
     
     if tracking_url_type_store != "file":
-        mlflow.sklearn.log_model(model, "s444421", registered_model_name="s444421", signature=signature, input_example=X_test.numpy()[:5])
+        mlflow.sklearn.log_model(model, "my_model", registered_model_name="s444421", signature=signature, input_example=X_test.numpy()[:5])
     else:
-        mlflow.sklearn.log_model(model, "s444421", signature=signature, input_example=X_test.numpy()[:5])
+        mlflow.sklearn.log_model(model, "my_model", signature=signature, input_example=X_test.numpy()[:5])
 
 
 # In[ ]: