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
@@ -19,7 +19,7 @@ def load_llama_model(model_path, is_guard=False):
 
     try:
         # Load tokenizer
-        tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL, token=HUGGINGFACE_TOKEN)
+        tokenizer = LlamaTokenizer.from_pretrained(BASE_MODEL, token=HUGGINGFACE_TOKEN)
 
         # Load config first (to avoid shape mismatch errors)
         config = AutoModelForCausalLM.from_pretrained(BASE_MODEL, config_only=True).config