Update app.py

app.py

@@ -186,15 +186,15 @@ if uploaded_image is not None:
                 with col7:
                     celeb = result[0]['class']
                     st.header("Predicted Image: ")
-                    if celeb == "MS Dhoni":
+                    if celeb == "ms_dhoni":
                         #dhoni = cv2.imread("dhoni.jpg")
                         dhoni = Image.open("dhoni.jpg")
                         st.image(dhoni,width=150, caption='MS Dhoni')
-                    elif celeb == "Rahul Dravid":
+                    elif celeb == "rahul_dravid":
                         #dravid = cv2.imread("rahul.jpg")
                         dravid = Image.open("rahul.jpg")
                         st.image(dravid,width=150, caption='Rahul Dravid')
-                    elif celeb == "Sachin Tendulkar":
+                    elif celeb == "sachin_tendulkar":
                         #sachin = cv2.imread("sachin.jpg")
                         sachin = Image.open("sachin.jpg")
                         st.image(sachin,width=150, caption='Sachin Tendulkar')