gradio/README.md

99 lines
3.9 KiB
Markdown
Raw Normal View History

2020-06-09 17:01:12 +08:00
[![CircleCI](https://circleci.com/gh/gradio-app/gradio-UI.svg?style=svg)](https://circleci.com/gh/gradio-app/gradio) [![PyPI version](https://badge.fury.io/py/gradio.svg)](https://badge.fury.io/py/gradio)
2020-06-02 03:54:27 +08:00
# Gradio UI
2019-09-18 02:05:54 +08:00
2020-06-27 01:13:15 +08:00
![alt text](https://i.ibb.co/GHRk2JP/header-2.png)
2020-06-23 06:27:14 +08:00
2020-06-12 03:44:44 +08:00
At Gradio, 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. We are working on making creating 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
2020-06-12 03:44:44 +08:00
To get a sense of `gradio`, take a look at the at the `examples` and `demo` folders, 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
2020-06-12 03:44:44 +08:00
Gradio is very easy to use with your existing code. Here's a 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
2020-06-12 03:44:44 +08:00
from imagenetlabels import idx_to_labels
2019-02-20 12:39:21 +08:00
2020-06-12 03:44:44 +08:00
def classify_image(inp):
2020-07-02 00:38:19 +08:00
inp = inp.reshape((1, 224, 224, 3))
prediction = mobile_net.predict(inp).flatten()
return {idx_to_labels[i].split(',')[0]: float(prediction[i]) for i in range(1000)}
2020-06-12 03:44:44 +08:00
2020-07-02 00:38:19 +08:00
imagein = gradio.inputs.Image(shape=(224, 224, 3))
2020-06-12 03:44:44 +08:00
label = gradio.outputs.Label(num_top_classes=3)
2020-07-02 00:38:19 +08:00
gr.Interface(classify_image, imagein, label, capture_session=True).launch();
2019-02-20 12:39:21 +08:00
```
2020-06-27 01:13:15 +08:00
![alt text](https://i.ibb.co/nM97z2B/image-interface.png)
2020-06-12 03:48:38 +08:00
2020-06-12 03:44:44 +08:00
You can supply your own model instead of the pretrained model above, as well as use different kinds of models or functions. 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
2020-06-12 03:44:44 +08:00
sketchpad = Sketchpad()
label = Label(num_top_classes=4)
2019-02-20 14:26:48 +08:00
2020-06-12 03:44:44 +08:00
gradio.Interface(predict, sketchpad, label).launch();
2019-02-20 14:26:48 +08:00
```
2020-06-27 01:13:15 +08:00
![alt text](https://i.ibb.co/CV8Kk3D/sketchpad-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
2020-06-12 03:45:17 +08:00
gradio.Interface(predict, 'textbox', 'label').launch()
2019-02-20 14:26:48 +08:00
```
2020-06-27 01:13:15 +08:00
![alt text](https://i.ibb.co/C7GXDDQ/label-interface.png)
2019-02-20 14:24:29 +08:00
2020-06-19 00:40:19 +08:00
### Contributing:
If you would like to contribute and your contribution is small, you can directly open a pull request (PR). If you would like to contribute a larger feature, we recommend first creating an issue with a proposed design for discussion. Please see our contributing guidelines for more info.
2020-06-12 03:44:44 +08:00
### See more:
Find more info on usage here: www.gradio.app.
2019-02-20 12:46:44 +08:00
2019-02-20 12:39:21 +08:00