import gradio as gr
from transformers import pipeline

title = 'Interactive Demo for Image Classification'
des = 'Demo for image classification using Vision Transformer built by Google Brain'
article = "<p style='text-align: center'><a href='https://www.linkedin.com/in/oayodeji/'>Space by Wvle</a> | <a href='https://arxiv.org/abs/2010.11929'>ViT</a>"

pipe = pipeline("image-classification")
gr.Interface.from_pipeline(pipe,
                           title=title,
                           description=des,
                           article=article,
                           examples=[['cat.jpg'], ['dog.jpg'], ['elephant.jpg']]
                           ).launch()