import streamlit as st
from PIL import Image, ImageOps
from transformers import pipeline

pipe = pipeline("image-classification", model="skyau/dog-breed-classifier-vit")
img = st.file_uploader("image")

if img:
    img = Image.open(img)
    label = pipe(img)
    st.success(f"Is it: {label}?")