Update app.py

app.py

@@ -1,13 +1,13 @@
 import torch
-from transformers import LlamaForCausalLM, LlamaTokenizer
+from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import LoraConfig, PeftModel
 import gradio as gr
 
 # Load tokenizer
-tokenizer = LlamaTokenizer.from_pretrained("VanguardAI/BhashiniLLaMa3-8B_LoRA_Adapters")
+tokenizer = AutoTokenizer.from_pretrained("VanguardAI/BhashiniLLaMa3-8B_LoRA_Adapters")
 
 # Load base model
-base_model = LlamaForCausalLM.from_pretrained("unsloth/llama-3-8b-Instruct-bnb-4bit")
+base_model = AutoModelForCausalLM.from_pretrained("unsloth/llama-3-8b-Instruct-bnb-4bit")
 
 # Apply LoRA adapters
 lora_config = LoraConfig(