ajcdp/Image-Classification-using-google-vit

1import gradio as gr2from transformers import pipeline3 4pipe = pipeline(task="image-classification", 5                model="google/vit-base-patch16-224")6 7 8gr.Interface.from_pipeline(pipe, 9                           title="Image Classification",10                           description="Please upload an image by clicking the 'Choose File' button below. We will use this image to demonstrate our image classification capabilities"11                           ).launch()