jye9's picture
Update app.py
4b5c69e verified
raw
history blame contribute delete
587 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline(task="image-classification",
# model that can do category classification
# https://huggingface.co/microsoft/beit-base-patch16-384
model = "microsoft/beit-base-patch16-384")
gr.Interface.from_pipeline(pipe,
title="Image Classification",
description="Object Recognition using Microsoft BEIT",
examples = ['egyptian_cat.jpg', 'German_shepherd.jpg',],
).launch()