explosion/spacy-models en_core_web_hftrf-3.8.1

on GitHub

Repackaged English transformer pipeline (roberta-base) for use with spacy-transformers>=1.4.0 and spacy>=3.8.0.

This is the en_core_web_trf 3.6.1 model with:

• Transformer weights converted from torch pickle to safetensors format
• Updated dependency pins for spacy 3.8.x and spacy-transformers 1.4.x
• Renamed to en_core_web_hftrf to avoid conflicts with the official model

Install

pip install https://github.com/explosion/spacy-models/releases/download/en_core_web_hftrf-3.8.1/en_core_web_hftrf-3.8.1-py3-none-any.whl

Usage

import spacy
nlp = spacy.load("en_core_web_hftrf")
doc = nlp("Apple is looking at buying U.K. startup for $1 billion")
print([(ent.text, ent.label_) for ent in doc.ents])
# [('Apple', 'ORG'), ('U.K.', 'GPE'), ('$1 billion', 'MONEY')]