Spaces:

tomsoderlund
/

swedish-entity-recognition

Runtime error

tomsoderlund commited on Dec 2, 2022

Commit

39fe662

1 Parent(s): 28dfd94

truncation=True

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ def process_swedish_text(text):
   # Models from https://huggingface.co/models
   # https://huggingface.co/KBLab/bert-base-swedish-cased-ner
   ml_model = 'KBLab/bert-base-swedish-cased-ner'
-  ner_pipeline = pipeline(model=ml_model, task='ner')
   # Run NER
   pipeline_results = ner_pipeline(text)
   print('NER results:', pipeline_results)

   # Models from https://huggingface.co/models
   # https://huggingface.co/KBLab/bert-base-swedish-cased-ner
   ml_model = 'KBLab/bert-base-swedish-cased-ner'
+  ner_pipeline = pipeline(model=ml_model, task='ner', truncation=True)
   # Run NER
   pipeline_results = ner_pipeline(text)
   print('NER results:', pipeline_results)