Georgi Gerganov
326573de9a
|
2 years ago | |
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| README.md | 2 years ago | |
| emscripten.cpp | 2 years ago | |
| index-tmpl.html | 2 years ago | |
README.md
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 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 language model
- The AI then vocalizes the response using the browser's 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.
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: