Upload README.md
Browse files
README.md
CHANGED
@@ -90,7 +90,8 @@ This is a fine-tuned GeoLM model for toponym detection task. The inputs are sent
|
|
90 |
|
91 |
To use this model, please refer to the code below.
|
92 |
|
93 |
-
* Option 1
|
|
|
94 |
```import torch
|
95 |
from transformers import AutoModelForTokenClassification, AutoTokenizer
|
96 |
|
@@ -122,7 +123,7 @@ predicted_labels = [model.config.id2label[label] for label in predicted_labels[0
|
|
122 |
# Print predicted labels
|
123 |
print(predicted_labels)
|
124 |
```
|
125 |
-
* Option 2
|
126 |
|
127 |
To appear soon
|
128 |
|
@@ -133,7 +134,8 @@ To appear soon
|
|
133 |
|
134 |
<!-- This should link to a Data Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
|
135 |
|
136 |
-
GeoWebNews (Credit to Gritta et al.)
|
|
|
137 |
Download link: https://github.com/milangritta/Pragmatic-Guide-to-Geoparsing-Evaluation/blob/master/data/GWN.xml
|
138 |
|
139 |
## Training Procedure
|
|
|
90 |
|
91 |
To use this model, please refer to the code below.
|
92 |
|
93 |
+
* **Option 1:** Load weights to a BERT model (Same procedure as the demo on the right side panel)
|
94 |
+
|
95 |
```import torch
|
96 |
from transformers import AutoModelForTokenClassification, AutoTokenizer
|
97 |
|
|
|
123 |
# Print predicted labels
|
124 |
print(predicted_labels)
|
125 |
```
|
126 |
+
* **Option 2:** Load weights to a GeoLM model
|
127 |
|
128 |
To appear soon
|
129 |
|
|
|
134 |
|
135 |
<!-- This should link to a Data Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
|
136 |
|
137 |
+
**GeoWebNews** (Credit to [Gritta et al.](https://arxiv.org/pdf/1810.12368.pdf))
|
138 |
+
|
139 |
Download link: https://github.com/milangritta/Pragmatic-Guide-to-Geoparsing-Evaluation/blob/master/data/GWN.xml
|
140 |
|
141 |
## Training Procedure
|