Commit
•
1f5fb2b
1
Parent(s):
2a0a46c
Update README.md
Browse files
README.md
CHANGED
@@ -20,7 +20,7 @@ from transformers import AutoModelForTokenClassification, AutoTokenizer
|
|
20 |
from nltk import word_tokenize
|
21 |
import torch
|
22 |
|
23 |
-
device = torch.device("cuda
|
24 |
path = f"KomeijiForce/roberta-large-metaie-super-academia-gpt4o"
|
25 |
tokenizer = AutoTokenizer.from_pretrained(path)
|
26 |
tagger = AutoModelForTokenClassification.from_pretrained(path).to(device)
|
|
|
20 |
from nltk import word_tokenize
|
21 |
import torch
|
22 |
|
23 |
+
device = torch.device("cuda")
|
24 |
path = f"KomeijiForce/roberta-large-metaie-super-academia-gpt4o"
|
25 |
tokenizer = AutoTokenizer.from_pretrained(path)
|
26 |
tagger = AutoModelForTokenClassification.from_pretrained(path).to(device)
|