Update app.py

app.py

@@ -17,8 +17,8 @@ DESCRIPTION = """
 # Welcome to Tonic's Grok-1
 """
 
-tokenizer = AutoTokenizer.from_pretrained(tokenizer_id, device_map="auto", trust_remote_code=True)
-# tokenizer = LlamaTokenizerFast.from_pretrained(tokenizer_id, device_map="cuda", trust_remote_code=True)
+# tokenizer = AutoTokenizer.from_pretrained(model_id, device_map="auto", trust_remote_code=True)
+tokenizer = LlamaTokenizerFast.from_pretrained(tokenizer_id, device_map="cuda", trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_id, load_in_4bit=True, device_map="cuda")
 
 def format_prompt(user_message, system_message="You are Grok-1, an AI language model created by Tonic-AI. You are a cautious assistant. You carefully follow instructions. You are helpful and harmless and follow ethical guidelines and promote positive behavior.\n\n"):