kfkas commited on
Commit
2f1f377
Β·
1 Parent(s): 76ee57c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -16,14 +16,14 @@ model-index:
16
 
17
  ## Usage
18
  ```python
19
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
20
  import torch
21
  import numpy as np
22
 
23
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
24
  model_dir = "kfkas/RoBERTa-large-Detection-P2G"
25
  tokenizer = AutoTokenizer.from_pretrained(model_dir)
26
- model = AutoModelForSeq2SeqLM.from_pretrained(model_dir)
27
 
28
  text = "μ›”λ“œμ»€ νŒŒλ‚˜μ€ν–‰ λŒ€ν‘œν‹°λ©” ν–‰μš°λŠ¬ μ΄λ‹¬λŸ¬ 이영영μž₯ μ„ λ¬Ό"
29
  with torch.no_grad():
 
16
 
17
  ## Usage
18
  ```python
19
+ from transformers import AutoTokenizer, RobertaForSequenceClassification
20
  import torch
21
  import numpy as np
22
 
23
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
24
  model_dir = "kfkas/RoBERTa-large-Detection-P2G"
25
  tokenizer = AutoTokenizer.from_pretrained(model_dir)
26
+ model = RobertaForSequenceClassification.from_pretrained(model_dir)
27
 
28
  text = "μ›”λ“œμ»€ νŒŒλ‚˜μ€ν–‰ λŒ€ν‘œν‹°λ©” ν–‰μš°λŠ¬ μ΄λ‹¬λŸ¬ 이영영μž₯ μ„ λ¬Ό"
29
  with torch.no_grad():