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
|