import gradio as gr
from transformers import pipeline
pipe = pipeline(task="image-classification",
                # model for 22k-category classification
                model="google/vit-base-patch16-224")
gr.Interface.from_pipeline(pipe,
                           title="VIT GOOGLE IMG CLASSIFIER",
                           description="Object Recognition ....",
                           examples = ['wonder_cat.jpg',
                                       'aki_dog.jpg',],
                           ).launch(inbrowser=True)