import gradio as gr
from transformers import pipeline

pipe = pipeline(task="image-classification", 
                model="google/vit-base-patch16-224")


gr.Interface.from_pipeline(pipe, 
                           title="Image Classification",
                           description="Please upload an image by clicking the 'Choose File' button below. We will use this image to demonstrate our image classification capabilities"
                           ).launch()