myworkspace / bw2cl.py
gauravml-89's picture
Update bw2cl.py
6658063 verified
raw
history blame contribute delete
543 Bytes
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,share=True)