| 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}?") | |