guild / app.py
pksh's picture
Create app.py
0d8c4c8
raw
history blame contribute delete
561 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline(task="image-classification",
# model for 22k-category classification
model="microsoft/beit-base-patch16-224-pt22k-ft22k")
gr.Interface.from_pipeline(pipe,
title="22k Image Classification",
description="Object Recognition ...",
examples = ['wonder_cat.jpg',
'aki_dog.jpg',],
).launch(inbrowser=True)