Update infer.py

infer.py

@@ -60,8 +60,9 @@ class PromptGuardAnomalyDetector(AbstractAnomalyDetector):
     def __init__(self, threshold: float):
         super().__init__(threshold)
         print('Loading prompt guard model...')
+        hf_token = os.environ.get('HF_TOKEN')
         self.classifier = pipeline(
-            "text-classification", model="meta-llama/Llama-Prompt-Guard-2-86M"
+            "text-classification", model="meta-llama/Llama-Prompt-Guard-2-86M", token=hf_token
         )
 
     def detect_anomaly(