sumitranjan
/

PromptShield

Text Classification

Model card Files Files and versions

sumitranjan commited on May 24, 2025

Commit

a0c2df9

·

verified ·

1 Parent(s): b8cb231

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -49,9 +49,9 @@ Whether you're building:
 ## 🧠 Model Architecture
-- **Base Model**: [`xlm-roberta-base`](https://huggingface.co/FacebookAI/xlm-roberta-base)
 - **Task**: Binary Sequence Classification
-- **Framework**: TensorFlow / Keras (`TFAutoModelForSequenceClassification`)
 - **Labels**:
   - `0` — Safe
   - `1` — Unsafe

 ## 🧠 Model Architecture
+- **Base Model**: FacebookAI/roberta-base
 - **Task**: Binary Sequence Classification
+- **Framework**: Pytorch
 - **Labels**:
   - `0` — Safe
   - `1` — Unsafe