Elron
/

bleurt-tiny-128

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Elron commited on Oct 4, 2021

Commit

1607b0b

•

1 Parent(s): e178a7b

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -19,7 +19,7 @@ references = ["hello world", "hello world"]
 candidates = ["hi universe", "bye world"]
 with torch.no_grad():
-  scores = model(**tokenizer(references, candidates, return_tensors='pt'))[0]
-print(scores) # tensor([[0.9877], [0.0475]])
 ```

 candidates = ["hi universe", "bye world"]
 with torch.no_grad():
+  scores = model(**tokenizer(references, candidates, return_tensors='pt'))[0].squeeze()
+print(scores) # tensor([-1.0563, -0.3004])
 ```