Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

av-codes
/
pi-detector-distilbert

Text Classification
Transformers
Safetensors
English
distilbert
prompt-injection
security
cybersecurity
llm-security
ml-intern
text-embeddings-inference
Model card Files Files and versions
xet
Community

Instructions to use av-codes/pi-detector-distilbert with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use av-codes/pi-detector-distilbert with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("text-classification", model="av-codes/pi-detector-distilbert")
    # Load model directly
    from transformers import AutoTokenizer, AutoModelForSequenceClassification
    
    tokenizer = AutoTokenizer.from_pretrained("av-codes/pi-detector-distilbert")
    model = AutoModelForSequenceClassification.from_pretrained("av-codes/pi-detector-distilbert")
  • Notebooks
  • Google Colab
  • Kaggle
pi-detector-distilbert
Ctrl+K
Ctrl+K
  • 1 contributor
History: 5 commits
av-codes's picture
av-codes
Update ML Intern artifact metadata
b870242 verified 2 days ago
  • .gitattributes
    1.52 kB
    initial commit 2 days ago
  • README.md
    3.16 kB
    Update ML Intern artifact metadata 2 days ago
  • config.json
    777 Bytes
    Best checkpoint - DistilBERT prompt injection detector, F1=1.0 val, 0.959 test 2 days ago
  • model.safetensors
    268 MB
    xet
    Best checkpoint - DistilBERT prompt injection detector, F1=1.0 val, 0.959 test 2 days ago
  • tokenizer.json
    711 kB
    Best checkpoint - DistilBERT prompt injection detector, F1=1.0 val, 0.959 test 2 days ago
  • tokenizer_config.json
    351 Bytes
    Best checkpoint - DistilBERT prompt injection detector, F1=1.0 val, 0.959 test 2 days ago
  • trainer_state.json
    2.87 kB
    Best checkpoint - DistilBERT prompt injection detector, F1=1.0 val, 0.959 test 2 days ago
  • training_args.bin

    Detected Pickle imports (10)

    • "transformers.training_args.OptimizerNames",
    • "transformers.trainer_utils.HubStrategy",
    • "accelerate.state.PartialState",
    • "transformers.trainer_utils.SchedulerType",
    • "transformers.trainer_pt_utils.AcceleratorConfig",
    • "torch.device",
    • "accelerate.utils.dataclasses.DistributedType",
    • "transformers.trainer_utils.SaveStrategy",
    • "transformers.training_args.TrainingArguments",
    • "transformers.trainer_utils.IntervalStrategy"

    How to fix it?

    5.33 kB
    xet
    Best checkpoint - DistilBERT prompt injection detector, F1=1.0 val, 0.959 test 2 days ago