Wootang01's picture
Update app.py
ec44c4c
raw
history blame
531 Bytes
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()