| import streamlit as st | |
| from PIL import Image, ImageOps | |
| from transformers import pipeline | |
| st.set_page_config( | |
| page_title="Dog Classifier", | |
| layout="centered", | |
| initial_sidebar_state="collapsed", | |
| ) | |
| banner_img = Image.open("siberian_husky.training_120476149_251778649544863_6165880252359897080_n-e1601592765529.jpg") | |
| st.image(banner_img) | |
| 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(str(label)) | |