| import streamlit as st | |
| from PIL import Image, ImageOps | |
| from transformers import pipeline | |
| st.set_page_config( | |
| page_title="Big Cat Classifier", | |
| layout="centered", | |
| initial_sidebar_state="collapsed", | |
| ) | |
| banner_img = Image.open("banner_img.png") | |
| 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)) | |