imvladikon commited on
Commit
0f55d8a
1 Parent(s): 522e605

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -43
README.md CHANGED
@@ -110,53 +110,11 @@ This is a [SpanMarker](https://github.com/tomaarsen/SpanMarkerNER) model trained
110
  from span_marker import SpanMarkerModel
111
 
112
  # Download from the 🤗 Hub
113
- model = SpanMarkerModel.from_pretrained("span_marker_model_id")
114
  # Run inference
115
  entities = model.predict("גרוסבורד נהג לבדו ב ה מכונית, ב דרכו מ ה עיר מיניאפוליס ב אינדיאנה ל נמל ה תעופה של היא.")
116
  ```
117
 
118
- ### Downstream Use
119
- You can finetune this model on your own dataset.
120
-
121
- <details><summary>Click to expand</summary>
122
-
123
- ```python
124
- from span_marker import SpanMarkerModel, Trainer
125
-
126
- # Download from the 🤗 Hub
127
- model = SpanMarkerModel.from_pretrained("span_marker_model_id")
128
-
129
- # Specify a Dataset with "tokens" and "ner_tag" columns
130
- dataset = load_dataset("conll2003") # For example CoNLL2003
131
-
132
- # Initialize a Trainer using the pretrained model & dataset
133
- trainer = Trainer(
134
- model=model,
135
- train_dataset=dataset["train"],
136
- eval_dataset=dataset["validation"],
137
- )
138
- trainer.train()
139
- trainer.save_model("span_marker_model_id-finetuned")
140
- ```
141
- </details>
142
-
143
- <!--
144
- ### Out-of-Scope Use
145
-
146
- *List how the model may foreseeably be misused and address what users ought not to do with the model.*
147
- -->
148
-
149
- <!--
150
- ## Bias, Risks and Limitations
151
-
152
- *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
153
- -->
154
-
155
- <!--
156
- ### Recommendations
157
-
158
- *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
159
- -->
160
 
161
  ## Training Details
162
 
 
110
  from span_marker import SpanMarkerModel
111
 
112
  # Download from the 🤗 Hub
113
+ model = SpanMarkerModel.from_pretrained("iahlt/span-marker-xlm-roberta-base-nemo-mt-he")
114
  # Run inference
115
  entities = model.predict("גרוסבורד נהג לבדו ב ה מכונית, ב דרכו מ ה עיר מיניאפוליס ב אינדיאנה ל נמל ה תעופה של היא.")
116
  ```
117
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
118
 
119
  ## Training Details
120