Commit
•
de467b0
1
Parent(s):
8681b05
Update README.md
Browse files
README.md
CHANGED
|
@@ -31,6 +31,7 @@ input = tokenizer(premise, hypothesis, truncation=True, return_tensors="pt")
|
|
| 31 |
output = model(input["input_ids"].to(device))[:,0] # device = "cuda:0" or "cpu"
|
| 32 |
prediction = torch.sigmoid(output).tolist()
|
| 33 |
print(prediction)
|
|
|
|
| 34 |
|
| 35 |
license: openrail
|
| 36 |
pipeline_tag: text-classification
|
|
|
|
| 31 |
output = model(input["input_ids"].to(device))[:,0] # device = "cuda:0" or "cpu"
|
| 32 |
prediction = torch.sigmoid(output).tolist()
|
| 33 |
print(prediction)
|
| 34 |
+
```
|
| 35 |
|
| 36 |
license: openrail
|
| 37 |
pipeline_tag: text-classification
|