Instructions to use Serj/intent-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Serj/intent-classifier with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("Serj/intent-classifier") model = AutoModelForSeq2SeqLM.from_pretrained("Serj/intent-classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -33,7 +33,7 @@ This is the model card of a 🤗 transformers model that has been pushed on the
|
|
| 33 |
|
| 34 |
|
| 35 |
## How to Get Started with the Model
|
| 36 |
-
|
| 37 |
class IntentClassifier:
|
| 38 |
def __init__(self, model_name="serj/intent-classifier", device="cuda"):
|
| 39 |
self.model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)
|
|
@@ -70,7 +70,7 @@ This is the model card of a 🤗 transformers model that has been pushed on the
|
|
| 70 |
"OPTIONS:\n refund\n cancel subscription\n damaged item\n return item\n", "Company",
|
| 71 |
"Products and subscriptions"))
|
| 72 |
|
| 73 |
-
|
| 74 |
[More Information Needed]
|
| 75 |
|
| 76 |
## Training Details
|
|
|
|
| 33 |
|
| 34 |
|
| 35 |
## How to Get Started with the Model
|
| 36 |
+
```
|
| 37 |
class IntentClassifier:
|
| 38 |
def __init__(self, model_name="serj/intent-classifier", device="cuda"):
|
| 39 |
self.model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)
|
|
|
|
| 70 |
"OPTIONS:\n refund\n cancel subscription\n damaged item\n return item\n", "Company",
|
| 71 |
"Products and subscriptions"))
|
| 72 |
|
| 73 |
+
```
|
| 74 |
[More Information Needed]
|
| 75 |
|
| 76 |
## Training Details
|