Upload model

README.md

@@ -1,21 +1,18 @@
+
 ---
 license: apache-2.0
-library_name: span_marker
+library_name: span-marker
 tags:
-- span_marker
+- span-marker
 - token-classification
 - ner
 - named-entity-recognition
 pipeline_tag: token-classification
-datasets:
-- DFKI-SLT/few-nerd
-language:
-- en
 ---
 
 # SpanMarker for Named Entity Recognition
 
-This is a [SpanMarker](https://github.com/tomaarsen/SpanMarkerNER) model that can be used for Named Entity Recognition. In particular, this SpanMarker model uses [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) as the underlying encoder.
+This is a [SpanMarker](https://github.com/tomaarsen/SpanMarkerNER) model that can be usedfor Named Entity Recognition. In particular, this SpanMarker model uses [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) as the underlying encoder. 
 
 ## Usage
 
@@ -25,15 +22,15 @@ To use this model for inference, first install the `span_marker` library:
 pip install span_marker
 ```
 
-You can then run inference as follows:
+You can then run inference with this model like so:
 
 ```python
 from span_marker import SpanMarkerModel
 
-# Download from Hub and run inference
-model = SpanMarkerModel.from_pretrained("tomaarsen/span-marker-bert-tiny-fewnerd-coarse-super")
+# Download from the 🤗 Hub
+model = SpanMarkerModel.from_pretrained("span_marker_model_name")
 # Run inference
 entities = model.predict("Amelia Earhart flew her single engine Lockheed Vega 5B across the Atlantic to Paris.")
 ```
 
-See the [SpanMarker](https://github.com/tomaarsen/SpanMarkerNER) repository for documentation and additional information on this model framework.
+See the [SpanMarker](https://github.com/tomaarsen/SpanMarkerNER) repository for documentation and additional information on this library.

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "models\\bt-coarse-fewnerd-1e-3-0\\checkpoint-final",
+  "_name_or_path": "models\\bt-full-coarse-1\\checkpoint-final",
   "architectures": [
     "SpanMarkerModel"
   ],
@@ -96,11 +96,12 @@
     "use_cache": true,
     "vocab_size": 30524
   },
-  "entity_max_length": 16,
-  "marker_max_length": 256,
-  "model_max_length": null,
+  "entity_max_length": 8,
+  "marker_max_length": 128,
+  "model_max_length": 256,
   "model_max_length_default": 512,
   "model_type": "span-marker",
+  "span_marker_version": "1.0.0.dev",
   "torch_dtype": "float32",
   "transformers_version": "4.27.2",
   "vocab_size": 30524

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:aaa2b68d313a8ee4d876d5240a251073b5b8168ea04e6578de32aaf43d099c5c
+oid sha256:7fad7460497ee6440ffe865221630a4d1b576a9cf192ed8ef9785a500e20ea0d
 size 17571375

tokenizer_config.json

@@ -4,7 +4,7 @@
   "do_basic_tokenize": true,
   "do_lower_case": true,
   "mask_token": "[MASK]",
-  "model_max_length": 256,
+  "model_max_length": 1000000000000000019884624838656,
   "never_split": null,
   "pad_token": "[PAD]",
   "sep_token": "[SEP]",