Navigace v modelech Hugging Face

Multi-Modal Models with Hugging Face

James Chapman

Curriculum Manager, DataCamp

Představení instruktora...

Sean Benson

Bývalý fyzik částic na Velkém hadronovém urychlovači (LHC) v CERN, Švýcarsko
Nyní aplikovaný výzkumný pracovník v oblasti AI na Amsterdamském univerzitním lékařském centru

Fotografie Seana Bensona.

Pokryté modality

Ukázky aplikací pro text, obrázky, zvuk a video.

¹ Image Credits: OpenAI generated

Hugging Face

Snímek obrazovky webové stránky modelu Hugging Face s informacemi z karty modelu

Více než 1 mil. modelů
Více než 250 tis. datových sad

Hub API

Programmatický přístup k modelům a datovým sadám

pip install huggingface_hub[cli]

Přihlaste se pro přístup k modelům ve vašem účtu:

>>> huggingface-cli login

Všechny knihovny, modely a datové sady budou předinstalovány

Vyhledávání modelů

Hugging Face API: programmatický přístup k modelům a datovým sadám

from huggingface_hub import HfApi
api = HfApi()

models = api.list_models()

task: "image-classification", "text-to-image" atd.
sort: např. "likes" nebo "downloads"
limit: maximální počet záznamů
tags: doplňující informace o modelu

Vyhledávání modelů

models = api.list_models(

    task="text-to-image",

    author="CompVis",

    tags="diffusers:StableDiffusionPipeline",

    sort="downloads"

)

top_model = list(models)[0]
print(top_model)

ModelInfo(id='CompVis/stable-diffusion-v1-4', private=False, downloads=1097285, 
          likes=6718, library_name='diffusers', ...

Použití modelů z API

top_model_id = top_model.id
print(top_model_id)

CompVis/stable-diffusion-v1-4

from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained(top_model_id)

Dostupné úlohy

Dokumentováno na webové stránce
Dostupné také ve formátu JSON: https://huggingface.co/api/tasks

import json
from urllib.request import urlopen
url = "https://huggingface.co/api/tasks"

with urlopen(url) as url:
    tasks = json.load(url)
print(tasks.keys())

dict_keys(['any-to-any',
'audio-classification',
'audio-to-audio', ...'])

¹ https://huggingface.co/tasks ² https://huggingface.co/api/tasks

Pojďme si procvičit!

Multi-Modal Models with Hugging Face