Commit
•
e090ac3
1
Parent(s):
8567f95
Update README.md
Browse files
README.md
CHANGED
|
@@ -23,7 +23,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
| 23 |
import torch
|
| 24 |
|
| 25 |
tokenizer = AutoTokenizer.from_pretrained("moranyanuka/icc")
|
| 26 |
-
model = AutoModelForSequenceClassification.from_pretrained("moranyanuka/icc).to("cuda")
|
| 27 |
|
| 28 |
captions = ["a great method of quantifying concreteness", "a man with a white shirt"]
|
| 29 |
text_ids = tokenizer(captions, padding=True, return_tensors="pt", truncation=True).to('cuda')
|
|
|
|
| 23 |
import torch
|
| 24 |
|
| 25 |
tokenizer = AutoTokenizer.from_pretrained("moranyanuka/icc")
|
| 26 |
+
model = AutoModelForSequenceClassification.from_pretrained("moranyanuka/icc").to("cuda")
|
| 27 |
|
| 28 |
captions = ["a great method of quantifying concreteness", "a man with a white shirt"]
|
| 29 |
text_ids = tokenizer(captions, padding=True, return_tensors="pt", truncation=True).to('cuda')
|