1 files changed, 18 insertions, 4 deletions
diff --git a/app/presenters/ml/model_presenter.rb b/app/presenters/ml/model_presenter.rb
index 388e2b73bc1..24d30af1d4e 100644
--- a/app/presenters/ml/model_presenter.rb
+++ b/app/presenters/ml/model_presenter.rb
@@ -5,17 +5,31 @@ module Ml
     presents ::Ml::Model, as: :model
 
     def latest_version_name
-      model.latest_version&.version
+      latest_version&.version
+    end
+
+    def version_count
+      return model.version_count if model.respond_to?(:version_count)
+
+      model.versions.size
     end
 
     def latest_package_path
-      return unless model.latest_version&.package_id.present?
+      latest_version&.package_path
+    end
 
-      Gitlab::Routing.url_helpers.project_package_path(model.project, model.latest_version.package_id)
+    def latest_version_path
+      latest_version&.path
     end
 
     def path
-      Gitlab::Routing.url_helpers.project_ml_model_path(model.project, model.id)
+      project_ml_model_path(model.project, model.id)
+    end
+
+    private
+
+    def latest_version
+      model.latest_version&.present
     end
   end
 end