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
|