Update app.py

app.py

@@ -19,7 +19,7 @@ FESTIVE = True  # Set to False for production-only mode
 
 print("🚀 Loading SAM-Z-1 Model...")
 
-MODEL_REPO = "Smilyai-labs/Sam-Z-1-tensorflow"
+MODEL_REPO = "Smilyai-labs/Sam-large-2"
 CACHE_DIR = "./model_cache"
 
 # ============================================================================
@@ -242,7 +242,7 @@ from transformers import AutoTokenizer
 hf_tokenizer = AutoTokenizer.from_pretrained("gpt2")
 
 # Add custom tokens to match model's vocab size
-custom_tokens = ["<|im_start|>", "<|im_end|>", "<think>", "<think/>"]
+custom_tokens = ["<|im_start|>", "<|im_end|>", "<think>", "<think/>", "<CONTINUE>"]
 hf_tokenizer.add_special_tokens({"additional_special_tokens": custom_tokens})
 
 # Save and reload as tokenizers format