diff --git a/src/AntSK/Pages/Setting/AIModel/ModelList.razor b/src/AntSK/Pages/Setting/AIModel/ModelList.razor
index e8c5f1f..c658df8 100644
--- a/src/AntSK/Pages/Setting/AIModel/ModelList.razor
+++ b/src/AntSK/Pages/Setting/AIModel/ModelList.razor
@@ -82,6 +82,10 @@
                                             {
                                                 <Tag Color="@PresetColor.Magenta.ToString()">StableDiffusion</Tag>
                                             }
+                                            else if (context.AIType == AIType.Ollama)
+                                            {
+                                                <Tag Color="@PresetColor.Pink.ToString()">Ollama</Tag>
+                                            }
                                         </p>
                                     </div>