Hugging Face's logo

sumitranjan
/
PromptShield

Text Classification
Keras
Safetensors
English
roberta
cybersecurity
llmsecurity
Model card Files
xet
Community
sumitranjan commited on
Commit
5d26e52
·
verified ·
1 Parent(s): d0187ca

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -48,7 +48,7 @@ from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
48
  import tensorflow as tf
49
 
50
  # Load tokenizer and model
51
- model_repo = "your-username/PromptShield"
52
  tokenizer = AutoTokenizer.from_pretrained(model_repo)
53
  model = TFAutoModelForSequenceClassification.from_pretrained(model_repo)
54
 
 
48
  import tensorflow as tf
49
 
50
  # Load tokenizer and model
51
+ model_repo = "sumitranjan/PromptShield"
52
  tokenizer = AutoTokenizer.from_pretrained(model_repo)
53
  model = TFAutoModelForSequenceClassification.from_pretrained(model_repo)
54