From 03a0978f5345e394522f21f1638d0d62f89d2796 Mon Sep 17 00:00:00 2001
From: NamH <NamNh0122@gmail.com>
Date: Thu, 14 Mar 2024 15:58:42 +0700
Subject: [PATCH] fix: some costmetic issues: badges corner, recommended for
 tensorrt models(#2346)

Signed-off-by: hiro <hiro@jan.ai>
Co-authored-by: hiro <hiro@jan.ai>
---
 uikit/src/badge/styles.scss                                | 2 +-
 web/containers/DropdownListSidebar/index.tsx               | 2 +-
 web/screens/ExploreModels/ExploreModelItemHeader/index.tsx | 7 ++++---
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/uikit/src/badge/styles.scss b/uikit/src/badge/styles.scss
index b777892d7..4788f65be 100644
--- a/uikit/src/badge/styles.scss
+++ b/uikit/src/badge/styles.scss
@@ -1,5 +1,5 @@
 .badge {
-  @apply focus:ring-ring border-border inline-flex items-center rounded-full border px-2.5 py-0.5 text-xs font-semibold transition-colors focus:outline-none focus:ring-2 focus:ring-offset-2;
+  @apply focus:ring-ring border-border inline-flex items-center rounded-md border px-2.5 py-0.5 text-xs font-semibold transition-colors focus:outline-none focus:ring-2 focus:ring-offset-2;
 
   &-primary {
     @apply border-transparent bg-blue-100 text-blue-600;
diff --git a/web/containers/DropdownListSidebar/index.tsx b/web/containers/DropdownListSidebar/index.tsx
index ef7532063..70651a4d4 100644
--- a/web/containers/DropdownListSidebar/index.tsx
+++ b/web/containers/DropdownListSidebar/index.tsx
@@ -294,7 +294,7 @@ const DropdownListSidebar = ({
                                 <span className="font-bold text-muted-foreground">
                                   {toGibibytes(x.metadata.size)}
                                 </span>
-                                {x.engine == InferenceEngine.nitro && (
+                                {x.metadata.size && (
                                   <ModelLabel size={x.metadata.size} />
                                 )}
                               </div>
diff --git a/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx b/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx
index 581a628ba..ef75f57f5 100644
--- a/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx
+++ b/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx
@@ -180,10 +180,11 @@ type EngineBadgeProps = {
 const EngineBadge: React.FC<EngineBadgeProps> = ({ engine }) => {
   switch (engine) {
     case 'nitro-tensorrt-llm':
+      const title = 'TensorRT-LLM'
       return (
-        <div className="flex items-center justify-center rounded-md bg-[#EFF6FF] px-2 py-[2px] font-semibold text-primary">
-          TensorRT-LLM
-        </div>
+        <Badge themes="primary" className="line-clamp-1" title={title}>
+          {title}
+        </Badge>
       )
     default:
       return null