iabachelis commited on
Commit
e7dfcd2
·
verified ·
1 Parent(s): 3bdcb4b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -67,12 +67,12 @@ using **Cross-Entropy** loss.
67
  ## How to Use
68
 
69
  ```python
70
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
71
  import torch
72
 
73
  m = "iabachelis/ModernBERT-TRABSA-CE"
74
  tok = AutoTokenizer.from_pretrained(m)
75
- model = AutoModelForSequenceClassification.from_pretrained(
76
  m, trust_remote_code=True).eval()
77
 
78
  text = "The film is visually stunning, but painfully slow."
 
67
  ## How to Use
68
 
69
  ```python
70
+ from transformers import AutoTokenizer, AutoModel
71
  import torch
72
 
73
  m = "iabachelis/ModernBERT-TRABSA-CE"
74
  tok = AutoTokenizer.from_pretrained(m)
75
+ model = AutoModel.from_pretrained(
76
  m, trust_remote_code=True).eval()
77
 
78
  text = "The film is visually stunning, but painfully slow."