From 9d8c705fd925b2f24e8c85231a46b303771601f9 Mon Sep 17 00:00:00 2001
From: Ettore Di Giacinto <mudler@users.noreply.github.com>
Date: Sun, 12 May 2024 14:24:36 +0200
Subject: [PATCH] feat(ui): display number of available models for installation
 (#2298)

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
---
 core/http/routes/ui.go      | 1 +
 core/http/views/models.html | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/core/http/routes/ui.go b/core/http/routes/ui.go
index 8cbb4b28..efd08315 100644
--- a/core/http/routes/ui.go
+++ b/core/http/routes/ui.go
@@ -85,6 +85,7 @@ func RegisterUIRoutes(app *fiber.App,
 			"Repositories":     appConfig.Galleries,
 			"AllTags":          tags,
 			"ProcessingModels": processingModelsData,
+			"AvailableModels":  len(models),
 			"TaskTypes":        taskTypes,
 			//	"ApplicationConfig": appConfig,
 		}
diff --git a/core/http/views/models.html b/core/http/views/models.html
index 44e16ef2..189e0356 100644
--- a/core/http/views/models.html
+++ b/core/http/views/models.html
@@ -10,7 +10,7 @@
 
         <div class="models mt-12">
             <h2 class="text-center text-3xl font-semibold text-gray-100">
-                🖼️ Available models from <i>{{ len .Repositories }}</i> repositories     <a href="https://localai.io/models/" target="_blank" >
+                🖼️ Available {{.AvailableModels}} models from <i>{{ len .Repositories }}</i> repositories     <a href="https://localai.io/models/" target="_blank" >
                     <i class="fas fa-circle-info pr-2"></i>
                 </a></h2>