Spaces:
Runtime error
Runtime error
import gradio as gr | |
from transformers import pipeline | |
pipe = pipeline(task="image-classification", | |
# model that can do 22k-category classification | |
model="abhishek/autotrain_fashion_mnist_vit_base") | |
gr.Interface.from_pipeline(pipe, | |
title="22k Image Classification", | |
description="Object Recognition using Microsoft BEIT", | |
examples = ['wonder_cat.jpg', 'aki_dog.jpg',], | |
article = "Author: <a href=\"https://huggingface.co/rowel\">Rowel Atienza</a>", | |
).launch() |