Hugging Face's logo

nvidia
/
domain-classifier

Transformers
Safetensors
deberta-v2
pytorch_model_hub_mixin
model_hub_mixin
Model card Files
xet
Community
6
sarahyurick commited on
Commit
e5d2151
·
verified ·
1 Parent(s): 41b2af6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -73,6 +73,7 @@ class CustomModel(nn.Module, PyTorchModelHubMixin):
73
  config = AutoConfig.from_pretrained("nvidia/domain-classifier")
74
  tokenizer = AutoTokenizer.from_pretrained("nvidia/domain-classifier")
75
  model = CustomModel.from_pretrained("nvidia/domain-classifier")
 
76
 
77
  # Prepare and process inputs
78
  text_samples = ["Sports is a popular domain", "Politics is a popular domain"]
 
73
  config = AutoConfig.from_pretrained("nvidia/domain-classifier")
74
  tokenizer = AutoTokenizer.from_pretrained("nvidia/domain-classifier")
75
  model = CustomModel.from_pretrained("nvidia/domain-classifier")
76
+ model.eval()
77
 
78
  # Prepare and process inputs
79
  text_samples = ["Sports is a popular domain", "Politics is a popular domain"]