gradio/README.md

102 lines
4.6 KiB
Markdown
Raw Normal View History

2019-09-18 02:05:54 +08:00
[![CircleCI](https://circleci.com/gh/gradio-app/gradio-UI.svg?style=svg)](https://circleci.com/gh/gradio-app/gradio)
2019-09-18 02:03:33 +08:00
# About Gradio
Gradio reduces the amount of data that a company needs to collect and label in order to build high-performing artificial intelligent systems.
The biggest bottleneck in building artificial intelligence systems is acquiring and labeling data. This process requires huge amounts of human labor and often requires domain expertise (e.g. costly lawyers to label contracts, costly doctors to label medical images). Applying recent machine learning research from Stanford, Gradio reduces the amount of data that an AI company needs to collect and label, often by as much as half.
This is achieved through: (1) identifying and correcting mislabelled data; (2) valuing each datapoint to automatically identify which data points should be labeled; and (3) intelligently augmenting data to diversify and maximize the value of the dataset. (See figure below.)
2019-09-18 02:08:09 +08:00
![Gradio modules](https://raw.githubusercontent.com/gradio-app/gradio-UI/master/screenshots/gradio-3-modules.PNG)
2019-09-18 02:03:33 +08:00
# Gradio UI Tool
2019-01-25 15:34:09 +08:00
2019-09-18 02:03:33 +08:00
As part of our work, we often try to understand what inputs that a model is particularly sensitive to. To help facilitate this, we've developed and open-sourced `gradio`, a python library that allows you to easily create input and output interfaces over trained models to make it easy for you to "play around" with your model in your browser by dragging-and-dropping in your own images (or pasting your own text, recording your own voice, etc.) and seeing what the model outputs. The library also automatically creates a shareable, public link to your model so you can share the interface with others (e.g. your client, your advisor, or your dad), who can use the model without writing any code.
2019-06-20 01:21:15 +08:00
Gradio is useful for:
2019-07-26 14:47:57 +08:00
* Creating demos of your machine learning code for clients / collaborators / users
* Getting feedback on model performance from users
* Debugging your model interactively during development
2019-06-20 01:18:07 +08:00
2019-06-20 01:18:33 +08:00
For more details, see the accompanying paper: ["Gradio: Hassle-Free Sharing and Testing of ML Models in the Wild"](https://arxiv.org/pdf/1906.02569.pdf), *ICML HILL 2019*, and please use the citation below.
2019-06-20 01:18:07 +08:00
```
@article{abid2019gradio,
title={Gradio: Hassle-Free Sharing and Testing of ML Models in the Wild},
author={Abid, Abubakar and Abdalla, Ali and Abid, Ali and Khan, Dawood and Alfozan, Abdulrahman and Zou, James},
journal={arXiv preprint arXiv:1906.02569},
year={2019}
}
```
2019-01-25 15:34:09 +08:00
2019-03-31 03:50:17 +08:00
To get a sense of `gradio`, take a look at the python notebooks in the `examples` folder, or read on below! And be sure to visit the gradio website: www.gradio.app.
2019-01-25 15:34:09 +08:00
2019-02-20 12:39:21 +08:00
## Installation
```
pip install gradio
```
(you may need to replace `pip` with `pip3` if you're running `python3`).
## Usage
2019-03-31 03:49:02 +08:00
Gradio is very easy to use with your existing code. Here is a minimum working example:
2019-02-20 12:39:21 +08:00
2019-02-20 14:26:48 +08:00
```python
2019-02-20 12:39:21 +08:00
import gradio
2019-03-31 03:49:02 +08:00
import tensorflow as tf
image_mdl = tf.keras.applications.inception_v3.InceptionV3()
2019-02-20 12:39:21 +08:00
2019-03-31 03:49:02 +08:00
io = gradio.Interface(inputs="imageupload", outputs="label", model_type="keras", model=image_mdl)
io.launch()
2019-02-20 12:39:21 +08:00
```
2019-03-31 03:49:02 +08:00
You can supply your own model instead of the pretrained model above, as well as use different kinds of models, not just keras models. Changing the `input` and `output` parameters in the `Interface` face object allow you to create different interfaces, depending on the needs of your model. Take a look at the python notebooks for more examples. The currently supported interfaces are as follows:
2019-02-20 12:39:21 +08:00
**Input interfaces**:
* Sketchpad
* ImageUplaod
* Webcam
* Textbox
**Output interfaces**:
2019-03-31 03:49:49 +08:00
* Label
2019-02-20 12:39:21 +08:00
* Textbox
## Screenshots
2019-02-20 12:46:44 +08:00
Here are a few screenshots that show examples of gradio interfaces
2019-03-31 03:49:49 +08:00
#### MNIST Digit Recognition (Input: Sketchpad, Output: Label)
2019-02-20 12:47:45 +08:00
2019-02-20 14:26:48 +08:00
```python
2019-03-31 03:49:49 +08:00
iface = gradio.Interface(input='sketchpad', output='label', model=model, model_type='keras')
2019-02-20 14:26:48 +08:00
iface.launch()
```
2019-07-12 03:57:29 +08:00
![alt text](https://raw.githubusercontent.com/abidlabs/gradio/master/screenshots/sketchpad_interface.png)
2019-02-20 14:24:29 +08:00
2019-07-12 03:57:29 +08:00
#### Image Classifier: InceptionNet (Input: Webcam, Output: Label)
2019-02-20 12:47:45 +08:00
2019-02-20 14:26:48 +08:00
```python
2019-03-31 03:49:49 +08:00
iface = gradio.Interface(inputs='webcam', outputs='label', model=model, model_type='keras')
2019-02-20 14:26:48 +08:00
iface.launch()
```
2019-07-12 03:57:29 +08:00
![alt text](https://raw.githubusercontent.com/abidlabs/gradio/master/screenshots/image_interface.png)
2019-02-20 14:24:29 +08:00
2019-07-12 03:57:29 +08:00
#### Human DNA Variant Effect Prediction (Input: Textbox, Output: Label)
2019-02-20 12:47:45 +08:00
2019-02-20 14:26:48 +08:00
```python
2019-03-31 03:49:49 +08:00
iface = gradio.Interface(inputs='textbox', outputs='label', model=model, model_type='keras')
2019-02-20 14:26:48 +08:00
iface.launch()
```
2019-07-12 03:57:29 +08:00
![alt text](https://raw.githubusercontent.com/abidlabs/gradio/master/screenshots/label_interface.png)
2019-02-20 14:24:29 +08:00
2019-11-02 00:53:58 +08:00
### What we're up to now:
Take a look at what we're working on now: www.gradio.app.
2019-02-20 12:46:44 +08:00
2019-02-20 12:39:21 +08:00