Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import torch
 import json
-from transformers import LlamaTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
 
 # Set Hugging Face Token for Authentication
@@ -35,7 +35,7 @@ def load_llama_model(model_path, is_guard=False):
             print("HUGGINGFACE_TOKEN found in environment")
         
         # Use the parameter name 'use_auth_token' instead of 'token'
-        tokenizer = LlamaTokenizer.from_pretrained(
+        tokenizer = AutoTokenizer.from_pretrained(
             BASE_MODEL,
             use_auth_token=token
         )