diff --git a/docs/evaluation/models.py b/docs/evaluation/models.py index c4348873..6845db16 100644 --- a/docs/evaluation/models.py +++ b/docs/evaluation/models.py @@ -4,11 +4,10 @@ from functools import partial +import dacy import spacy from spacy.language import Language -import dacy - def scandiner_loader() -> Language: scandiner = spacy.blank("da") @@ -117,7 +116,8 @@ def openai_model_loader_fine_ner(model: str) -> Language: "da_core_news_md-3.5.0": partial(spacy.load, "da_core_news_md"), "da_core_news_sm-3.5.0": partial(spacy.load, "da_core_news_sm"), "openai/gpt-3.5-turbo (02/05/23)": partial( - openai_model_loader_simple_ner, model="gpt-3.5-turbo" + openai_model_loader_simple_ner, + model="gpt-3.5-turbo", ), "openai/gpt-4 (02/05/23)": partial(openai_model_loader_simple_ner, model="gpt-4"), }