File size: 531 Bytes
ada2446
b28f796
 
ec44c4c
 
b28f796
 
92679b1
b28f796
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
import gradio as gr
from transformers import pipeline

title = "Image Classifier"
description = " To use it, simply add your image, or click one of the examples to load them. Read more at the links below."

pipe = pipeline(task="image-classification", 
                model="microsoft/beit-base-patch16-224-pt22k-ft22k")
gr.Interface.from_pipeline(pipe, 
                           title=title,
                           description=description,
                           enable_queue=True,
                           ).launch()