Spaces:
Sleeping
Sleeping
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() |