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
@@ -26,7 +26,7 @@ LLAMA_GUARD_NAME = "meta-llama/Llama-Guard-3-1B-INT4"  # Ensure this is correct
 def load_llama_model(base_model=BASE_MODEL, adapter=None):
     print(f"🔄 Loading Base Model: {base_model}")
     
-    tokenizer = AutoTokenizer.from_pretrained(base_model, token=HUGGINGFACE_TOKEN)
+    tokenizer = LlamaTokenizer.from_pretrained(base_model, token=HUGGINGFACE_TOKEN)
     model = AutoModelForCausalLM.from_pretrained(
         base_model,
         token=HUGGINGFACE_TOKEN,