[demo] Fix token-classification (Closes huggingface#116)

ocavue · May 19, 2023 · 4a6b8cc · 4a6b8cc
1 parent 8d8f511
commit 4a6b8cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/demo-site/src/worker.js b/examples/demo-site/src/worker.js
@@ -292,7 +292,7 @@ async function token_classification(data) {
     let currentChunk = { type: '', text: [] };
 
     for (let i = 0; i < outputs.length; i++) {
-        let word = pipeline.tokenizer.model.tokens_to_ids[outputs[i].word];
+        let word = pipeline.tokenizer.model.tokens_to_ids.get(outputs[i].word);
         let entity = outputs[i].entity;
 
         if (entity.startsWith('B-')) { // beginning of a new chunk