HuggingFace Hub image generation example not working

[jtpio]

Description

Trying to run the following example notebook gives an error: https://github.com/jupyterlab/jupyter-ai/blob/main/examples/images.ipynb

Reproduce

pip install jupyterlab jupyter-ai huggingface_hub

export HUGGINGFACEHUB_API_TOKEN=abcd1234

jupyter lab

Then run the following cell in a notebook:

%%ai huggingface_hub:stabilityai/stable-diffusion-2-1 --format image
It's an astronaut with a boombox

Gives the following error:

ValueError: Task unknown has no recommended model. Please specify a model explicitly. Visit https://huggingface.co/tasks for more info.

Expected behavior

It should produce an image.

Context

• Jupyter AI 2.29.1
• huggingface_hub==0.28.1

[jtpio]

Of note, maybe it could be worth looking into switching to langchain-huggingface: https://python.langchain.com/docs/integrations/providers/huggingface/

Maybe this would help fix the deprecation warning:

FutureWarning: 'post' (from 'huggingface_hub.inference._client') is deprecated and will be removed from version '0.31.0'. Making direct POST requests to the inference server is not supported anymore. Please use task methods instead (e.g. `InferenceClient.chat_completion`)