Hugging Face's logo

mochodek
/
comment-bert-purpose

Text Classification
Transformers
PyTorch
English
bert
text-embeddings-inference
Model card Files
xet
 Community
1
mochodek commited on
Commit
0b4a0ed
·
verified ·
1 Parent(s): 9f7b679

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -24,7 +24,7 @@ encoded_input = tokenizer(text, return_tensors='pt')
24
 
25
  output = model(**encoded_input)
26
 
27
- scores = softmax(output.logits.cpu().detach().numpy())
28
 
29
  id2class[np.argmax(scores)]
30
 
 
24
 
25
  output = model(**encoded_input)
26
 
27
+ scores = softmax(output.logits.detach().numpy())
28
 
29
  id2class[np.argmax(scores)]
30