Modelpublic

proventra/mdeberta-v3-base-prompt-injection

sourceHugging Facemitupdated 1y agoView on Hugging Face

3likes3.9kdownloads

Model Card

mdeberta-v3-base-prompt-injection

This model is a fine-tuned version of microsoft/mdeberta-v3-base on a combination of jackhhao/jailbreak-classification, deepset/prompt-injections, a custom datasets containing known attacks, and injections nested in legitimate content like websites and articles.

Usage

Python

from transformers import pipeline

classifier = pipeline(
  "text-classification",
  model="proventra/mdeberta-v3-base-prompt-injection"
)

print(classifier("Your text to scan"))

Use in Proventra Core

proventra-core python library

check out Proventra