KomeijiForce
/

roberta-large-metaie-super-academia-gpt4o

Token Classification

Inference Endpoints

Model card Files Files and versions Community

KomeijiForce commited on May 24

Commit

1f5fb2b

•

1 Parent(s): 2a0a46c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -20,7 +20,7 @@ from transformers import AutoModelForTokenClassification, AutoTokenizer
 from nltk import word_tokenize
 import torch
-device = torch.device("cuda:1")
 path = f"KomeijiForce/roberta-large-metaie-super-academia-gpt4o"
 tokenizer = AutoTokenizer.from_pretrained(path)
 tagger = AutoModelForTokenClassification.from_pretrained(path).to(device)

 from nltk import word_tokenize
 import torch
+device = torch.device("cuda")
 path = f"KomeijiForce/roberta-large-metaie-super-academia-gpt4o"
 tokenizer = AutoTokenizer.from_pretrained(path)
 tagger = AutoModelForTokenClassification.from_pretrained(path).to(device)