Update app.py

app.py

@@ -1,10 +1,15 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import os
 
-# Load the Llama model
+# Set Hugging Face token
+# Replace this with your Hugging Face token or set it as an environment variable in Spaces settings.
+hf_token = os.getenv("HF_TOKEN")
+
+# Load the Llama model with authentication
 model_name = "meta-llama/Llama-3.2-3B-Instruct"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
+model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_token)
 
 # Define chatbot function
 def chat_with_llama(user_input):