classify-pets / app.py
miittnnss's picture
Create app.py
5c852b7
raw
history blame contribute delete
242 Bytes
import gradio as gr
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-classification", model="miittnnss/pet-classifier-v2")
iface = gr.Interface.from_pipeline(pipe)
iface.queue()
iface.launch()