File size: 587 Bytes
4b5c69e
349e4c9
 
4b5c69e
 
 
 
 
349e4c9
4b5c69e
 
349e4c9
1
2
3
4
5
6
7
8
9
10
11
12
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()