import spacy_streamlit from pathlib import Path import srsly import importlib import random DEFAULT_MODEL = "en_core_web_sm" DEFAULT_TEXT = "David Bowie moved to the US in 1974, initially staying in New York City before settling in Los Angeles." DESCRIPTION = """**Explore trained [spaCy v3.1](https://nightly.spacy.io) pipelines**""" def get_all_models(): with open("requirements.txt") as f: content = f.readlines() models = [] for line in content: if "huggingface.co" in line: models.append(line.split("/")[4]) return models MODELS = get_all_models() def get_default_text(nlp): # Check if spaCy has built-in example texts for the language try: examples = importlib.import_module(f".lang.{nlp.lang}.examples", "spacy") return examples.sentences[0] except (ModuleNotFoundError, ImportError): return "" spacy_streamlit.visualize( MODELS, default_model=DEFAULT_MODEL, visualizers=["parser", "ner", "similarity", "tokens"], show_visualizer_select=True, sidebar_description=DESCRIPTION, get_default_text=get_default_text )