Update infer.py

infer.py

@@ -61,7 +61,7 @@ class PromptGuardAnomalyDetector(AbstractAnomalyDetector):
         super().__init__(threshold)
         print('Loading prompt guard model...')
         self.classifier = pipeline(
-            "text-classification", model="../data/models/Prompt-Guard-86M"
+            "text-classification", model="meta-llama/Llama-Prompt-Guard-2-86M"
         )
 
     def detect_anomaly(
@@ -80,7 +80,7 @@ class PromptGuardAnomalyDetector(AbstractAnomalyDetector):
                 KnownAttackVector(
                     known_prompt=anomaly["label"],
                     similarity_percentage=anomaly["score"],
-                    source="Prompt-Guard-86M",
+                    source="meta-llama/Llama-Prompt-Guard-2-86M",
                 )
                 for anomaly in anomalies
                 if anomaly["score"] >= threshold