README.md

# KinoBumblebee

[![Docs](https://img.shields.io/badge/hex.pm-docs-8e7ce6.svg)](https://hexdocs.pm/kino_bumblebee)
[![Actions Status](https://github.com/livebook-dev/kino_bumblebee/workflows/Test/badge.svg)](https://github.com/livebook-dev/kino_bumblebee/actions)

[Bumblebee](https://github.com/elixir-nx/bumblebee) integration for [Livebook](https://livebook.dev).

This ships by default as part of Livebook v0.8.0. Simply [install Livebook](https://livebook.dev/#install),
open up a new notebook, and choose the "Neural Networks task" under the "+ Smart" cell prompt.

## Installation

To bring KinoBumblebee to Livebook all you need to do is `Mix.install/2`:

```elixir
Mix.install([
  {:kino_bumblebee, "~> 0.3.0"}
])
```

## Examples

After installing this package in Livebook you can interactively experiment
with a number of Machine-Learning tasks, using different pre-trained models.
A couple examples of what's possible:

### Text-to-image

![](.github/images/text_to_image.png)

### Image classification

![](.github/images/image_classification.png)

### Image-to-text

![](.github/images/image_to_text.png)

### Text classification

![](.github/images/text_classification.png)

### Token classification

![](.github/images/token_classification.png)

### Zero-shot text classification

![](.github/images/zero_shot_text_classification.png)

### Fill-mask

![](.github/images/fill_mask.png)

### Question answering

![](.github/images/question_answering.png)

### Text generation

![](.github/images/text_generation.png)

### Speech-to-text

![](.github/images/speech_to_text.png)

### Conversation

![](.github/images/conversation.png)


## License

Copyright (C) 2022 Dashbit

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at [http://www.apache.org/licenses/LICENSE-2.0](http://www.apache.org/licenses/LICENSE-2.0)

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.