Update app.py

app.py

@@ -15,7 +15,7 @@ __class_number_to_name = {}
 __model = None
 
 st.header("Welcome to Indian Cricketers Classifier!")
-col1,col2,col3 = st.columns(3)
+col1,col2,col3, col4 = st.columns(3)
 with col1: 
     #dhoni = cv2.imread("dhoni.jpg")
     dhoni = Image.open("dhoni.jpg")
@@ -40,6 +40,13 @@ with col3:
     #sehwag = cv2.imread("sehwag.jpg")
     sehwag = Image.open("sehwag.jpg")
     st.image(sehwag,width=150, caption='Virendra Sehwag')
+with col4:
+    sunil_gavaskar = Image.open("sunil_gavaskar.jpg")
+    st.image(sunil_gavaskar,width=150, caption='Sunil Gavaskar')
+
+    #sehwag = cv2.imread("sehwag.jpg")
+    kapil_dev = Image.open("kapil_dev.jpg")
+    st.image(kapil_dev,width=150, caption='Kapil Dev')
 
     
 
@@ -129,13 +136,13 @@ def load_saved_artifacts():
     global __class_name_to_number
     global __class_number_to_name
 
-    with open("class_cri_dictionary.json", "r") as f:
+    with open("class_cri_dictionary1.json", "r") as f:
         __class_name_to_number = json.load(f)
         __class_number_to_name = {v:k for k,v in __class_name_to_number.items()}
 
     global __model
     if __model is None:
-        __model = joblib.load('cri_saved_model.pkl') 
+        __model = joblib.load('cri_saved_model1.pkl') 
     #st.text("loading saved artifacts...done")
     return True
 
@@ -203,6 +210,12 @@ if uploaded_image is not None:
                         #sehwag = cv2.imread("sehwag.jpg")
                         sehwag = Image.open("sehwag.jpg")
                         st.image(sehwag,width=150, caption='Virendra Sehwag')
+                    elif celeb == "sunil_gavaskar":
+                        sunil_gavaskar = Image.open("sunil_gavaskar.jpg")
+                        st.image(sunil_gavaskar,width=150, caption='Sunil Gavaskar')
+                    elif celeb == "kapil_dev":
+                        kapil_dev = Image.open("kapil_dev.jpg")
+                        st.image(kapil_dev,width=150, caption='Kapil Dev')
             except:
                 st.header("Image Cannot be Classified!Please Try Again")