Hub

Использование Flair на Hugging Face

Flair — это очень простой фреймворк для передовых методов NLP. Разработан Берлинским университетом Гумбольдта и друзьями.

Исследование Flair в Hub

Модели flair можно найти, используя фильтр слева на странице моделей.

Все модели на Hub обладают следующими полезными возможностями:

Автоматически сгенерированная карточка модели с кратким описанием.
Интерактивный виджет, с помощью которого можно взаимодействовать с моделью непосредственно в браузере.
Виджет Inference Providers, позволяющий отправлять запросы на инференс.

Для начала вы можете воспользоваться руководством по установке Flair. Также можно установить пакет одной строкой через pip:

$ pip install -U flair

Все модели flair можно легко загрузить с Hub:

from flair.data import Sentence
from flair.models import SequenceTagger

# load tagger
tagger = SequenceTagger.load("flair/ner-multi")

После загрузки вы можете использовать predict() для выполнения инференса:

sentence = Sentence("George Washington ging nach Washington.")
tagger.predict(sentence)

# print sentence
print(sentence)

Выводится следующее:

Sentence[6]: "George Washington ging nach Washington." → ["George Washington"/PER, "Washington"/LOC]

Если вы хотите загрузить конкретную модель Flair, нажмите Use in Flair на карточке модели — и вы получите готовый фрагмент кода!