5hadytru commited on
Commit
4ba3345
·
1 Parent(s): 31688d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
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