Update app.py

app.py

@@ -210,7 +210,7 @@ config = None
 eos_token_id = None
 fast_forward = None
 
-MODEL_REPO = "Smilyai-labs/Sam-Z-1-tensorflow"
+MODEL_REPO = "Smilyai-labs/Sam-1-large-it-0002"
 CACHE_DIR = "./model_cache"
 
 # Stats
@@ -336,11 +336,11 @@ def format_chat_prompt(messages: List[ChatMessage]) -> str:
     prompt = ""
     for msg in messages:
         if msg.role == "user":
-            prompt += f"<|im_start|>user\n{msg.content}<|im_end|>\n"
+            prompt += f"User: {msg.content}"
         elif msg.role == "assistant":
-            prompt += f"<|im_start|>assistant\n{msg.content}<|im_end|>\n"
+            prompt += f"Sam: {msg.content}"
     
-    prompt += "<|im_start|>assistant\n"
+    prompt += "Sam: "
     return prompt
 
 # ============================================================================
@@ -783,7 +783,7 @@ async def load_model():
         from transformers import AutoTokenizer
         
         hf_tokenizer = AutoTokenizer.from_pretrained("gpt2")
-        custom_tokens = ["<|im_start|>", "<|im_end|>", "<think>", "<think/>"]
+        custom_tokens = ["<think>", "<think/>"]
         hf_tokenizer.add_special_tokens({"additional_special_tokens": custom_tokens})
         
         os.makedirs("./temp_tokenizer", exist_ok=True)