Update app.py

app.py

@@ -1,11 +1,21 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import login
 
 # Load your fine-tuned model
 model_name = "ayeshaNoor1/Llama_finetunedModel-1b"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
+
+# Authenticate with the access token
+login("HF_ACCESS_TOKEN")
+
+model_name = "ayeshaNoor1/Llama_finetunedModel-1b"
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=True)
+
+
 # Define the text generation function
 def generate_text(prompt):
     inputs = tokenizer(prompt, return_tensors="pt")