From e7e3a09ce4fe0e6fa6aa0140f53e049fa6e641bf Mon Sep 17 00:00:00 2001
From: DarokCx <77368869+DarokCx@users.noreply.github.com>
Date: Tue, 1 Oct 2024 13:01:12 -0400
Subject: [PATCH] bugfixes

---
 public/index.html                |  6 +++---
 public/scripts/textgen-models.js | 13 +++++++++----
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/public/index.html b/public/index.html
index cbef346d4..db0f77b3f 100644
--- a/public/index.html
+++ b/public/index.html
@@ -2276,7 +2276,7 @@
                                             </select>
                                             <select id="category_selection">
                                                 <option value="" disabled selected data-i18n="category">category</option>
-                                                <option value="Favorite" data-i18n="Top">Favorite</option>
+                                                <!-- <option value="Favorite" data-i18n="Top">Favorite</option> -->
                                                 <option value="Top" data-i18n="Top">Top</option>
                                                 <option value="New" data-i18n="New">New</option>
                                                 <option value="All" data-i18n="All">All</option>
@@ -2291,11 +2291,11 @@
 
                                     </div>
                                 </div>
-                                <select id="featherless_model">
+                                <!-- <select id="featherless_model">
                                     <option value="" data-i18n="-- Connect to the API --">
                                         -- Connect to the API --
                                     </option>
-                                </select>
+                                </select> -->
                             </div>
                             <div data-tg-type="vllm">
                                 <div class="flex-container flexFlowColumn">
diff --git a/public/scripts/textgen-models.js b/public/scripts/textgen-models.js
index 8398b358b..dd7c58793 100644
--- a/public/scripts/textgen-models.js
+++ b/public/scripts/textgen-models.js
@@ -412,11 +412,16 @@ export async function loadFeatherlessModels(data) {
         const selectedSortOrder = sortOrderSelect.value;
         const selectedClass = classSelect.value;
         const selectedCategory = categoriesSelect.value;
+        let featherlessTop = [];
+        let featherlessNew = [];
 
-        const featherlessTop = await fetchFeatherlessStats();
+        if (selectedCategory === 'Top') {
+            featherlessTop = await fetchFeatherlessStats();
+        }
         const featherlessIds = featherlessTop.map(stat => stat.id);
-
-        const featherlessNew = await fetchFeatherlessNew();
+        if(selectedCategory === 'New') {
+            featherlessNew = await fetchFeatherlessNew();
+        }
         const featherlessNewIds = featherlessNew.map(stat => stat.id);
 
         let filteredModels = originalModels.filter(model => {
@@ -435,7 +440,7 @@ export async function loadFeatherlessModels(data) {
                 return matchesSearch && matchesClass && matchesNew;
             }
             else {
-                return null;
+                return matchesSearch;
             }
         });