diff --git a/src/views/components/InputMessage/index.tsx b/src/views/components/InputMessage/index.tsx
index d8aaf45..2779663 100644
--- a/src/views/components/InputMessage/index.tsx
+++ b/src/views/components/InputMessage/index.tsx
@@ -248,11 +248,9 @@ const InputMessage = observer((props: any) => {
   const getModelShowName = (modelName: string) => {
     const nameMap = {
       "gpt-3.5-turbo": "GPT-3.5",
-      "gpt-3.5-turbo-1106": "GPT-3.5-1106",
-      "gpt-3.5-turbo-16k": "GPT-3.5-16K",
       "gpt-4": "GPT-4",
       "gpt-4-1106-preview": "GPT-4-turbo",
-      "claude-2": "CLAUDE-2",
+      "claude-2.1": "CLAUDE 2.1",
     };
     if (modelName in nameMap) {
       return nameMap[modelName];