Update README.md (#1)
Browse files- Update README.md (834a26179e61bf7ccece66267377ea09a01b732f)
Co-authored-by: Vibhu Jawa <[email protected]>
README.md
CHANGED
|
@@ -109,6 +109,7 @@ class CustomModel(nn.Module, PyTorchModelHubMixin):
|
|
| 109 |
config = AutoConfig.from_pretrained("nvidia/multilingual-domain-classifier")
|
| 110 |
tokenizer = AutoTokenizer.from_pretrained("nvidia/multilingual-domain-classifier")
|
| 111 |
model = CustomModel.from_pretrained("nvidia/multilingual-domain-classifier")
|
|
|
|
| 112 |
|
| 113 |
# Prepare and process inputs
|
| 114 |
text_samples = ["Los deportes son un dominio popular", "La política es un dominio popular"]
|
|
|
|
| 109 |
config = AutoConfig.from_pretrained("nvidia/multilingual-domain-classifier")
|
| 110 |
tokenizer = AutoTokenizer.from_pretrained("nvidia/multilingual-domain-classifier")
|
| 111 |
model = CustomModel.from_pretrained("nvidia/multilingual-domain-classifier")
|
| 112 |
+
model.eval()
|
| 113 |
|
| 114 |
# Prepare and process inputs
|
| 115 |
text_samples = ["Los deportes son un dominio popular", "La política es un dominio popular"]
|