Add token-based access for gated model

app.py

@@ -1,10 +1,12 @@
+import os
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 model_id = "meta-llama/Llama-3.1-8B-Instruct"
+hf_token = os.environ.get("HF_TOKEN")  # Hugging Face access token (added via Secrets)
 
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token)
+model = AutoModelForCausalLM.from_pretrained(model_id, token=hf_token, device_map="auto")
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 def bust_myth(statement):