Update app.py

app.py

@@ -21,19 +21,24 @@ with col1:
     dhoni = Image.open("dhoni.jpg")
     st.image(dhoni,width=150, caption='MS Dhoni')
 
-    ganguly = cv2.imread("ganguly.jpg")
+    #ganguly = cv2.imread("ganguly.jpg")
+    ganguly = Image.open("ganguly.jpg")
     st.image(ganguly,width=150, caption='Saurav Ganguly')
 with col2:
-    dravid = cv2.imread("rahul.jpg")
-    st.image(dravid,width=150, caption='Rahul Dravid')
+    #rahul = cv2.imread("rahul.jpg")
+    rahul = Image.open("rahul.jpg")
+    st.image(rahul,width=150, caption='Rahul Dravid')
 
-    virat = cv2.imread("virat.jpg")
+    #virat = cv2.imread("virat.jpg")
+    virat = Image.open("virat.jpg")
     st.image(virat,width=150, caption='Virat Kohli')
 with col3:
-    sachin = cv2.imread("sachin.jpg")
+    #sachin = cv2.imread("sachin.jpg")
+    sachin = Image.open("sachin.jpg")
     st.image(sachin,width=150, caption='Sachin Tendulkar')
 
-    sehwag = cv2.imread("sehwag.jpg")
+    #sehwag = cv2.imread("sehwag.jpg")
+    sehwag = Image.open("sehwag.jpg")
     st.image(sehwag,width=150, caption='Virendra Sehwag')
 
     
@@ -174,23 +179,29 @@ if uploaded_image is not None:
                     celeb = result[0]['class']
                     st.header("Predicted Image: ")
                     if celeb == "MS Dhoni":
-                        dhoni = cv2.imread("dhoni.jpg")
+                        #dhoni = cv2.imread("dhoni.jpg")
+                        dhoni = Image.open("dhoni.jpg")
                         st.image(dhoni,width=150, caption='MS Dhoni')
                     elif celeb == "Rahul Dravid":
-                        dravid = cv2.imread("rahul.jpg")
+                        #dravid = cv2.imread("rahul.jpg")
+                        dravid = Image.open("rahul.jpg")
                         st.image(dravid,width=150, caption='Rahul Dravid')
                     elif celeb == "Sachin Tendulkar":
-                        sachin = cv2.imread("sachin.jpg")
+                        #sachin = cv2.imread("sachin.jpg")
+                        sachin = Image.open("sachin.jpg")
                         st.image(sachin,width=150, caption='Sachin Tendulkar')
                     elif celeb == "Saurav Ganguly":
-                        ganguly = cv2.imread("ganguly.jpg")
+                        #ganguly = cv2.imread("ganguly.jpg")
+                        ganguly = Image.open("ganguly.jpg")
                         st.image(ganguly,width=150, caption='Saurav Ganguly')
                     elif celeb == "virat_kohli":
-                        virat = cv2.imread("virat.jpg")
+                        #virat = cv2.imread("virat.jpg")
+                        virat = Image.open("virat.jpg")
                         st.image(virat,width=150, caption='Virat Kohli')
                     elif celeb == "Virendra Sehwag":
-                        sehwag = cv2.imread("sehwag.jpg")
-                        st.image(virat,width=150, caption='Virendra Sehwag')
+                        #sehwag = cv2.imread("sehwag.jpg")
+                        sehwag = Image.open("sehwag.jpg")
+                        st.image(sehwag,width=150, caption='Virendra Sehwag')
             except:
                 st.header("Image Cannot be Classified!")