Update README.md

README.md

@@ -33,7 +33,9 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 model = AutoModelForSequenceClassification.from_pretrained('HeTree/HeCross')
 tokenizer = AutoTokenizer.from_pretrained('HeTree/HeCross')
-features = tokenizer(['כמה אנשים חיים בברלין?', 'כמה אנשים חיים בברלין?'], ['ברלין מונה 3,520,031 תושבים רשומים בשטח של 891.82 קמ"ר.', 'העיר ניו יורק מפורסמת בזכות מוזיאון המטרופוליטן לאומנות.'],  padding=True, truncation=True, return_tensors="pt")
+features = tokenizer(['כמה אנשים חיים בברלין?', 'כמה אנשים חיים בברלין?'],
+                     ['ברלין מונה 3,520,031 תושבים רשומים בשטח של 891.82 קמ"ר.', 'העיר ניו יורק מפורסמת בזכות מוזיאון המטרופוליטן לאומנות.'],
+                     padding=True, truncation=True, return_tensors="pt")
 model.eval()
 with torch.no_grad():
     scores = sigmoid(model(**features).logits)