nvidia
/

multilingual-domain-classifier

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions Community

Update README.md

#1

by VibhuJawa - opened 9 days ago

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -109,6 +109,7 @@ class CustomModel(nn.Module, PyTorchModelHubMixin):
 config = AutoConfig.from_pretrained("nvidia/multilingual-domain-classifier")
 tokenizer = AutoTokenizer.from_pretrained("nvidia/multilingual-domain-classifier")
 model = CustomModel.from_pretrained("nvidia/multilingual-domain-classifier")
 # Prepare and process inputs
 text_samples = ["Los deportes son un dominio popular", "La política es un dominio popular"]

 config = AutoConfig.from_pretrained("nvidia/multilingual-domain-classifier")
 tokenizer = AutoTokenizer.from_pretrained("nvidia/multilingual-domain-classifier")
 model = CustomModel.from_pretrained("nvidia/multilingual-domain-classifier")
+model.eval()
 # Prepare and process inputs
 text_samples = ["Los deportes son un dominio popular", "La política es un dominio popular"]