Instructions to use Moaaz2os/stip-threat-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Moaaz2os/stip-threat-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Moaaz2os/stip-threat-classifier")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Moaaz2os/stip-threat-classifier") model = AutoModelForSequenceClassification.from_pretrained("Moaaz2os/stip-threat-classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload BertForSequenceClassification
Browse files- config.json +1 -0
- model.safetensors +1 -1
config.json
CHANGED
|
@@ -34,6 +34,7 @@
|
|
| 34 |
"num_hidden_layers": 12,
|
| 35 |
"pad_token_id": 0,
|
| 36 |
"position_embedding_type": "absolute",
|
|
|
|
| 37 |
"torch_dtype": "float32",
|
| 38 |
"transformers_version": "4.40.0",
|
| 39 |
"type_vocab_size": 2,
|
|
|
|
| 34 |
"num_hidden_layers": 12,
|
| 35 |
"pad_token_id": 0,
|
| 36 |
"position_embedding_type": "absolute",
|
| 37 |
+
"problem_type": "single_label_classification",
|
| 38 |
"torch_dtype": "float32",
|
| 39 |
"transformers_version": "4.40.0",
|
| 40 |
"type_vocab_size": 2,
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 436367360
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:682bfe3015ffcfdc1604ca9d724c837211498b8881608ba1fb361c71767b9b1c
|
| 3 |
size 436367360
|