Update README.md

README.md

@@ -16,9 +16,14 @@ library_name: transformers
 
 ```python
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
-checkpoint = "david-inf/bert-sci-am"
 
-model = AutoModelForSequenceClassification.from_pretrained(
-    "david-inf/bert-sci-am", num_labels=2)
-tokenizer = AutoTokenizer.from_pretrained("david-inf/bert-sci-am")
+def load_model():
+    """Load model from hub"""
+    checkpoint = "david-inf/bert-sci-am"
+    model = AutoModelForSequenceClassification.from_pretrained(
+        checkpoint, num_labels=2)
+    tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+    return model, tokenizer
+
+model, tokenizer = load_model()
 ```