Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import torch
 import json
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import LlamaTokenizer, AutoModelForCausalLM
 from peft import PeftModel
 
 # Set Hugging Face Token for Authentication
@@ -21,7 +21,7 @@ LLAMA_GUARD_NAME = "meta-llama/Llama-Guard-3-1B-INT4"
 def load_llama_model(model_name, is_guard=False):
     print(f"🔄 Loading Model: {model_name}")
     
-    tokenizer = AutoTokenizer.from_pretrained(model_name, token=HUGGINGFACE_TOKEN)
+    tokenizer = LlamaTokenizer.from_pretrained(model_name, token=HUGGINGFACE_TOKEN)
     model = AutoModelForCausalLM.from_pretrained(
         model_name,
         token=HUGGINGFACE_TOKEN,