Update app.py

app.py

@@ -2,15 +2,16 @@ import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import os
 
+# Retrieve the Hugging Face token from environment variables
 HF_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
 
 if HF_TOKEN is None:
     raise ValueError("HUGGINGFACE_TOKEN is not set. Add it in your Space's secrets.")
 
-# Load model and tokenizer
+# Load model and tokenizer with authentication
 MODEL_NAME = "meta-llama/Llama-Guard-3-1B"
-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)
 
 # Streamlit UI
 st.title("AI Safe Content Checker Tool")