Update README.md

README.md

@@ -33,6 +33,10 @@ Identify the section of a legal contract that contains the governing law and ext
 
 Use the code below to get started with the model.
 ```python
-from clf_ner import clf_ner
-clf_ner.predict("This agreement shall be governed by the laws of the State of New Jersey")
+from transformers import AutoTokenizer, ClassifierNER
+
+tokenizer = AutoTokenizer.from_pretrained("sguarnaccio/gov_law_clf_ner")
+
+model = ClassifierNER.from_pretrained("sguarnaccio/gov_law_clf_ner")
+model.predict("This agreement shall be governed by the laws of the State of New Jersey")
 ```