https://github.com/livebook-dev/kino_bumblebee

Bumblebee (pre-trained neural networks) integration for Livebook
https://github.com/livebook-dev/kino_bumblebee

Last synced: 7 days ago
JSON representation

Bumblebee (pre-trained neural networks) integration for Livebook

• Host: GitHub
• URL: https://github.com/livebook-dev/kino_bumblebee
• Owner: livebook-dev
• License: apache-2.0
• Created: 2022-11-29T20:13:33.000Z (almost 2 years ago)
• Default Branch: main
• Last Pushed: 2024-06-17T16:26:48.000Z (5 months ago)
• Last Synced: 2024-09-21T14:14:29.316Z (about 2 months ago)
• Language: Elixir
• Homepage:
• Size: 3.68 MB
• Stars: 94
• Watchers: 9
• Forks: 13
• Open Issues: 2
• Metadata Files:
  • Readme: README.md
  • Changelog: CHANGELOG.md
  • License: LICENSE

Awesome Lists containing this project

• awesome-livebook - Bumblebee - Bumblebee (pre-trained neural networks) integration for Livebook - _Tags: Machine Learning_  (SmartCells)
• awesome-livebook - Bumblebee - Bumblebee (pre-trained neural networks) integration for Livebook - _Tags: Machine Learning_  (SmartCells)

README

        
# 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.5.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.