Update README.md
Browse files
README.md
CHANGED
|
@@ -98,6 +98,8 @@ predicted_labels = [model.config.id2label[label] for label in predicted_labels[0
|
|
| 98 |
|
| 99 |
# Print predicted labels
|
| 100 |
print(predicted_labels)
|
|
|
|
|
|
|
| 101 |
```
|
| 102 |
* **Option 2:** Load weights to a GeoLM model
|
| 103 |
|
|
|
|
| 98 |
|
| 99 |
# Print predicted labels
|
| 100 |
print(predicted_labels)
|
| 101 |
+
# ['O', 'B-Topo', 'O', 'O', 'O', 'O', 'O', 'B-Topo', 'O', 'O', 'O', 'O', 'O', 'O',
|
| 102 |
+
# 'O', 'O', 'B-Topo', 'O', 'O', 'O', 'O', 'O', 'B-Topo', 'I-Topo', 'I-Topo', 'O', 'O', 'O']
|
| 103 |
```
|
| 104 |
* **Option 2:** Load weights to a GeoLM model
|
| 105 |
|