Update README.md

README.md

@@ -25,7 +25,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 model_id = "oshizo/japanese-sexual-moderation-v2"
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
-model_ = AutoModelForSequenceClassification.from_pretrained(
+model = AutoModelForSequenceClassification.from_pretrained(
     model_id,
     problem_type="regression"
 )
@@ -38,7 +38,7 @@ texts = [
 
 with torch.no_grad():
     encoding = tokenizer(texts, padding=True, max_length=64, return_tensors="pt")
-    scores = model_(**encoding).logits
+    scores = model(**encoding).logits
 
 # tensor([[-0.0045],
 #         [ 0.2954],