Pass HF_TOKEN to pipeline for private model access

app.py

@@ -4,13 +4,15 @@ Space: https://huggingface.co/spaces/Builder117/Prompt_Injection
 Model: Builder117/distilbert-prompt-injection (fine-tuned DistilBERT)
 """
 
+import os
 import gradio as gr
 from transformers import pipeline
 
 MODEL_ID = "Builder117/distilbert-prompt-injection"
+HF_TOKEN = os.environ.get("HF_TOKEN")
 
 print(f"Loading model: {MODEL_ID} ...")
-classifier = pipeline("text-classification", model=MODEL_ID, device=-1)
+classifier = pipeline("text-classification", model=MODEL_ID, device=-1, token=HF_TOKEN)
 print("Model ready.")