Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,14 @@ import streamlit as st
|
|
2 |
from PIL import Image, ImageOps
|
3 |
from transformers import pipeline
|
4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
pipe = pipeline("image-classification", model="skyau/dog-breed-classifier-vit")
|
6 |
img = st.file_uploader("image")
|
7 |
|
|
|
2 |
from PIL import Image, ImageOps
|
3 |
from transformers import pipeline
|
4 |
|
5 |
+
st.set_page_config(
|
6 |
+
page_title="Big Cat Classifier",
|
7 |
+
layout="centered",
|
8 |
+
initial_sidebar_state="collapsed",
|
9 |
+
)
|
10 |
+
banner_img = Image.open("banner_img.png")
|
11 |
+
st.image(banner_img)
|
12 |
+
|
13 |
pipe = pipeline("image-classification", model="skyau/dog-breed-classifier-vit")
|
14 |
img = st.file_uploader("image")
|
15 |
|