Update README.md

README.md

@@ -57,4 +57,14 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification
 
 tokenizer = AutoTokenizer.from_pretrained("botryan96/GeoBERT")
 model = AutoModelForTokenClassification.from_pretrained("botryan96/GeoBERT")
+
+#Define the pipeline
+from transformers import pipeline
+ner_machine = pipeline('ner',model = models,tokenizer=tokenizer, aggregation_strategy="simple")
+
+#Define the sentence
+sentence = 'In North America, the water storage in the seepage face model is higher than the base case because positive pore pressure is requisite for drainage through a seepage face boundary condition. The result from the resistivity data supports the notion, especially in the northern part of the Sandstone Sediment formation. The active formation of America has a big potential for Oil and Gas based on the seismic section, has been activated since the Paleozoic'
+
+#Deploy the NER Machine
+ner_machine(sentence)
 ```