Update README.md

README.md

@@ -14,9 +14,10 @@ class TemporalRelationClassificationPipeline(TextClassificationPipeline):
     def check_model_type(self, supported_models):
         pass
 
-model = AutoModelForSequenceClassification.from_pretrained("guyyanko/split-3-hebrew-trc-alephbert-base-EMP",
-                                                           trust_remote_code=True)
-tokenizer = AutoTokenizer.from_pretrained("split-1-hebrew-trc-alephbert-base-EMP", trust_remote_code=True)
+pretrained_checkpoint = "guyyanko/split-3-hebrew-trc-alephbert-base-EMP"
+
+model = AutoModelForSequenceClassification.from_pretrained(pretrained_checkpoint, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained(pretrained_checkpoint, trust_remote_code=True)
 classifier = pipeline(task='text-classification', model=model, tokenizer=tokenizer)
 
 txt = "מחר [א1] אתאמן [/א1] אם [א2] אסיים [/א2] את כל המשימות שלי"