Update README.md
Browse files
README.md
CHANGED
@@ -50,4 +50,15 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
50 |
|
51 |
tokenizer = AutoTokenizer.from_pretrained("botryan96/GeoBERT_analyzer")
|
52 |
model = AutoModelForTokenClassification.from_pretrained("botryan96/GeoBERT_analyzer")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
53 |
```
|
|
|
50 |
|
51 |
tokenizer = AutoTokenizer.from_pretrained("botryan96/GeoBERT_analyzer")
|
52 |
model = AutoModelForTokenClassification.from_pretrained("botryan96/GeoBERT_analyzer")
|
53 |
+
|
54 |
+
#Define the pipeline
|
55 |
+
from transformers import pipeline
|
56 |
+
anlyze_machine=pipeline('text-classification',model = model_checkpoint2)
|
57 |
+
|
58 |
+
#Define the sentences
|
59 |
+
sentences = ['the average iron and sulfate concentrations were calculated to be 19 . 6 5 . 2 and 426 182 mg / l , respectively .',
|
60 |
+
'She first gained media attention as a friend and stylist of Paris Hilton']
|
61 |
+
|
62 |
+
#Deploy the machine
|
63 |
+
anlyze_machine(sentences)
|
64 |
```
|