Skip to content
Create a web-based UI around your machine learning model in 3 minutes
Branch: master
Clone or download
Latest commit 297f91e Jul 12, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.circleci more fixes Jun 19, 2019
build/lib/gradio version 0.7.8 Jun 22, 2019
dist version 0.7.8 Jun 22, 2019
examples just removed notebooks Apr 29, 2019
gradio.egg-info version 0.7.8 Jun 22, 2019
gradio Merge branch 'master' of github.com:abidlabs/gradiome into abidlabs/a… Jun 22, 2019
gradiohub added gradiohub Jun 11, 2019
screenshots add screenshots Jul 11, 2019
src saliency Apr 29, 2019
test more fixes Jun 19, 2019
web added gradiohub Jun 11, 2019
.gitignore just removed notebooks Apr 29, 2019
LICENSE Create LICENSE Jul 10, 2019
MANIFEST.in 0.4.4 Mar 25, 2019
README.md Update README.md Jul 11, 2019
build-interface.py fix all interfaces Apr 10, 2019
index.html seperated inputs and outputs, added webcam interface Feb 17, 2019
setup.py version 0.7.8 Jun 22, 2019
test.png saliency Apr 29, 2019

README.md

Gradio

CircleCI

Gradio is 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. Gradio also 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.

Gradio is useful for:

  • Creating demos for clients
  • Getting feedback from collaborators
  • Debugging your model during development

For more details, see the accompanying paper: "Gradio: Hassle-Free Sharing and Testing of ML Models in the Wild", ICML HILL 2019, and please use the citation below.

@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}
}

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.

Installation

pip install gradio

(you may need to replace pip with pip3 if you're running python3).

Usage

Gradio is very easy to use with your existing code. Here is a minimum working example:

import gradio
import tensorflow as tf
image_mdl = tf.keras.applications.inception_v3.InceptionV3()

io = gradio.Interface(inputs="imageupload", outputs="label", model_type="keras", model=image_mdl)
io.launch()

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:

Input interfaces:

  • Sketchpad
  • ImageUplaod
  • Webcam
  • Textbox

Output interfaces:

  • Label
  • Textbox

Screenshots

Here are a few screenshots that show examples of gradio interfaces

MNIST Digit Recognition (Input: Sketchpad, Output: Label)

iface = gradio.Interface(input='sketchpad', output='label', model=model, model_type='keras')
iface.launch()

alt text

Image Classifier: InceptionNet (Input: Webcam, Output: Label)

iface = gradio.Interface(inputs='webcam', outputs='label', model=model, model_type='keras')
iface.launch()

alt text

Human DNA Variant Effect Prediction (Input: Textbox, Output: Label)

iface = gradio.Interface(inputs='textbox', outputs='label', model=model, model_type='keras')
iface.launch()

alt text

More Documentation

More detailed and up-to-date documentation can be found on the gradio website: www.gradio.app.

You can’t perform that action at this time.