nvidia
/

domain-classifier

pytorch_model_hub_mixin

model_hub_mixin

Inference Endpoints

Model card Files Files and versions Community

sarahyurick commited on 13 days ago

Commit

e5d2151

•

1 Parent(s): 41b2af6

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -73,6 +73,7 @@ class CustomModel(nn.Module, PyTorchModelHubMixin):
 config = AutoConfig.from_pretrained("nvidia/domain-classifier")
 tokenizer = AutoTokenizer.from_pretrained("nvidia/domain-classifier")
 model = CustomModel.from_pretrained("nvidia/domain-classifier")
 # Prepare and process inputs
 text_samples = ["Sports is a popular domain", "Politics is a popular domain"]

 config = AutoConfig.from_pretrained("nvidia/domain-classifier")
 tokenizer = AutoTokenizer.from_pretrained("nvidia/domain-classifier")
 model = CustomModel.from_pretrained("nvidia/domain-classifier")
+model.eval()
 # Prepare and process inputs
 text_samples = ["Sports is a popular domain", "Politics is a popular domain"]