Aller au contenu

Créer une interface web avec Gradio#

Gradio est un package Python qui permet de créer des interfaces web pour les modèles de machine learning.

Gradio vous permet de créer rapidement des composants d'interface utilisateur personnalisables autour de vos modèles TensorFlow ou PyTorch, ou même des fonctions Python arbitraires. Mélangez et associez les composants pour prendre en charge toute combinaison d'entrées et de sorties.....

Les composants de Gradio#

Une application Gradio est composée de trois composantes

  • des entrées
  • une fonction de traitement
  • des sorties

Le fonctionnement de Gradio est relativement simplement, on fournit des entrées qui vont être traitées par la fonction de traitement et retourner des sorties.

Installation#

Gradio peut être installé depuis le répertoire PyPi. 

pip install gradio

Première application#

Créer un fichier app.py

import gradio as gr

Créer la fonction de traitement#

def greet(name):
    return f"Hello {name} !"

Définir les entrées et les sorties#

input_name = gr.Textbox(label="Name")
output_msg = gr.Textbox(label="Message")

app = gr.Interface(fn=greet,
                   inputs=input_name,
                   outputs=output_msg,
                   title="Greeting App",
                   description="Enter your name to get a greeting message")

Lancer l'application#

app.launch()

Le code complet#

app_greet.py
import gradio as gr


def greet(name):
    return f"Hello {name} !"


input_name = gr.Textbox(label="Name")
output_msg = gr.Textbox(label="Message")

app = gr.Interface(fn=greet,
                   inputs=input_name,
                   outputs=output_msg,
                   title="Greeting App",
                   description="Enter your name to get a greeting message")

app.launch()

Image Black & White Converter#

Créer un fichier app.py

import gradio as gr

Créer la fonction de traitement#

def greet(name):
    return f"Hello {name} !"

Définir les entrées et les sorties#

input_name = gr.Textbox(label="Name")
output_msg = gr.Textbox(label="Message")

app = gr.Interface(fn=greet,
                   inputs=input_name,
                   outputs=output_msg,
                   title="Greeting App",
                   description="Enter your name to get a greeting message")

Lancer l'application#

app.launch()

Le code complet#

app_greet.py
# Packages
import gradio as gr
from PIL import Image


# Serving Function
def convert_image(image):
    pil_image = Image.fromarray(image)
    pil_image = pil_image.convert("L")
    return pil_image


# Setup interface
input_image = gr.Image()
output_image = gr.Image()
app = gr.Interface(fn=convert_image, 
                   inputs=input_image, 
                   outputs=output_image,
                   title="Image Black & White Converter",
                   description="Simple tool to convert any image in Black & White")

# Launch
app.launch()

Modèle de classification d'image#

Créons à pésent une interface pour un modèle de classification d'image

Pour en savoir plus, merci de consulter la documentation de Gradio ici https://gradio.app/.