|
|
|
|
# talk.wasm
|
|
|
|
|
|
|
|
|
|
Talk with an Artificial Intelligence in your browser:
|
|
|
|
|
|
|
|
|
|
https://user-images.githubusercontent.com/1991296/203411580-fedb4839-05e4-4474-8364-aaf1e9a9b615.mp4
|
|
|
|
|
|
|
|
|
|
Online demo: https://talk.ggerganov.com
|
|
|
|
|
|
|
|
|
|
## How it works?
|
|
|
|
|
|
|
|
|
|
This demo leverages 2 modern neural network models to create a high-quality voice chat directly in your browser:
|
|
|
|
|
|
|
|
|
|
- [OpenAI's Whisper](https://github.com/openai/whisper) speech recognition model is used to process your voice and understand what you are saying
|
|
|
|
|
- Upon receiving some voice input, the AI generates a text response using [OpenAI's GPT-2](https://github.com/openai/gpt-2) language model
|
|
|
|
|
- The AI then vocalizes the response using the browser's [Web Speech API](https://developer.mozilla.org/en-US/docs/Web/API/Web_Speech_API)
|
|
|
|
|
|
|
|
|
|
The web page does the processing locally on your machine. The processing of these heavy neural network models in the
|
|
|
|
|
browser is possible by implementing them efficiently in C/C++ and using the browser's WebAssembly SIMD capabilities for
|
|
|
|
|
extra performance. For more detailed information, checkout the [current repository](https://github.com/ggerganov/whisper.cpp).
|
|
|
|
|
|
|
|
|
|
In order to run the models, the web page first needs to download the model data which is about ~350 MB. The model data
|
|
|
|
|
is then cached in your browser's cache and can be reused in future visits without downloading it again.
|
|
|
|
|
|
|
|
|
|
## Requirements
|
|
|
|
|
|
|
|
|
|
In order to run this demo efficiently, you need to have the following:
|
|
|
|
|
|
|
|
|
|
- Latest Chrome or Firefox browser (Safari is not supported)
|
|
|
|
|
- Run this on a desktop or laptop with modern CPU (a mobile phone will likely not be good enough)
|
|
|
|
|
- Speak phrases that are no longer than 10 seconds - this is the audio context of the AI
|
|
|
|
|
- The web-page uses about 1.4GB of RAM
|
|
|
|
|
|
|
|
|
|
Notice that this demo is using the smallest GPT-2 model, so the generated text responses are not always very good.
|
|
|
|
|
Also, the prompting strategy can likely be improved to achieve better results.
|
|
|
|
|
|
|
|
|
|
The demo is quite computationally heavy - it's not usual to run these transformer models in a browser. Typically, they
|
|
|
|
|
run on powerful GPU hardware. So for better experience, you do need to have a powerful computer.
|
|
|
|
|
|
|
|
|
|
Probably in the near future, mobile browsers will start supporting WASM SIMD. This will allow to run the demo on your
|
|
|
|
|
phone or tablet. But for now this functionality is not supported on mobile devices (at least not on iPhone).
|
|
|
|
|
|
|
|
|
|
## Todo
|
|
|
|
|
|
|
|
|
|
- Better UI (contributions are welcome)
|
|
|
|
|
- Better GPT-2 prompting
|
|
|
|
|
|
|
|
|
|
## Feedback
|
|
|
|
|
|
|
|
|
|
If you have any comments or ideas for improvement, please drop a comment in the following discussion:
|
|
|
|
|
|
|
|
|
|
https://github.com/ggerganov/whisper.cpp/discussions/167
|
