Update app.py

app.py

@@ -1,8 +1,20 @@
 import tensorflow as tf
 model=tf.keras.models.load_model('model.h5')
 import streamlit as st
-st.header("Adidas VS Nike Classification")
-categories=['Adidas','NIKE']
+st.header("Wonderful Wonders Classification")
+st.markdown("This model takes in the input image of any wonder of the world and tries to classify it.")
+categories=['Roman Colosseum',
+ 'Stonehenge',
+ 'Machu Pichu',
+ 'Chichen Itza',
+ 'Christ The Reedemer',
+ 'Eiffel Tower',
+ 'Taj Mahal',
+ 'Pyramids Of Giza',
+ 'Statue of Liberty',
+ 'Burj Khalifa',
+ 'Venezuela Angel Falls',
+ 'Great Wall of China']
 from PIL import Image
 uploaded_image=st.file_uploader("Upload image",type=["jpg","jpeg","webp"])
 import numpy as np
@@ -17,4 +29,4 @@ if(uploaded_image!=None):
         img=img.reshape(1,150,150,3)
         pred=model.predict(img)
         print(pred[0][0])
-        st.text(categories[round(pred[0][0])])
+        st.text(categories[np.argmax(pred)])