Instructions to use daxa-ai/pebblo-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use daxa-ai/pebblo-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="daxa-ai/pebblo-classifier")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("daxa-ai/pebblo-classifier") model = AutoModelForSequenceClassification.from_pretrained("daxa-ai/pebblo-classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update config.json
Browse files- config.json +9 -9
config.json
CHANGED
|
@@ -36,14 +36,6 @@
|
|
| 36 |
"label2id": {
|
| 37 |
"Board meeting": 0,
|
| 38 |
"Consulting Agreement": 1,
|
| 39 |
-
"Customer List": 2,
|
| 40 |
-
"Distribution/Partner Agreement": 3,
|
| 41 |
-
"Enterprise License Agreement": 4,
|
| 42 |
-
"Executive Severance Agreement": 5,
|
| 43 |
-
"Financial Report": 6,
|
| 44 |
-
"HARMFUL_ADVICE": 7,
|
| 45 |
-
"Internal Use Only": 8,
|
| 46 |
-
"Loan and security Agreement": 9,
|
| 47 |
"MEDICAL_ADVICE": 10,
|
| 48 |
"Merger Agreement": 11,
|
| 49 |
"NDA": 12,
|
|
@@ -54,8 +46,16 @@
|
|
| 54 |
"Security Breach": 17,
|
| 55 |
"Settlement Agreement": 18,
|
| 56 |
"Sexual Harrassment": 19,
|
|
|
|
| 57 |
"employee agreement": 20,
|
| 58 |
-
"enterprise agreement": 21
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
},
|
| 60 |
"max_position_embeddings": 512,
|
| 61 |
"model_type": "distilbert",
|
|
|
|
| 36 |
"label2id": {
|
| 37 |
"Board meeting": 0,
|
| 38 |
"Consulting Agreement": 1,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
"MEDICAL_ADVICE": 10,
|
| 40 |
"Merger Agreement": 11,
|
| 41 |
"NDA": 12,
|
|
|
|
| 46 |
"Security Breach": 17,
|
| 47 |
"Settlement Agreement": 18,
|
| 48 |
"Sexual Harrassment": 19,
|
| 49 |
+
"Customer List": 2,
|
| 50 |
"employee agreement": 20,
|
| 51 |
+
"enterprise agreement": 21,
|
| 52 |
+
"Distribution/Partner Agreement": 3,
|
| 53 |
+
"Enterprise License Agreement": 4,
|
| 54 |
+
"Executive Severance Agreement": 5,
|
| 55 |
+
"Financial Report": 6,
|
| 56 |
+
"HARMFUL_ADVICE": 7,
|
| 57 |
+
"Internal Use Only": 8,
|
| 58 |
+
"Loan and security Agreement": 9
|
| 59 |
},
|
| 60 |
"max_position_embeddings": 512,
|
| 61 |
"model_type": "distilbert",
|