Update app.py

app.py

@@ -163,10 +163,35 @@ if uploaded_image is not None:
             img = cv2.resize(img, (130, 130))
             
             result = classify_image(get_b64_test_image_for_virat(), display_image)
+           
             try:
-                st.text(result[0])
-            except:
-                st.text("Image Cannot be Classified!")
+                col6,col7 = st.columns(2)
+                with col6:
+                    st.header("Uploded Image: ")
+                    st.image(img,width=130, caption='Uploaded Image')
+                with col7:
+                    celeb = result[0]['class']
+                    st.header("Predicted Image: ")
+                    if celeb == "MS Dhoni":
+                        dhoni = cv2.imread("dhoni.jpg")
+                        st.image(dhoni,width=150, caption='MS Dhoni')
+                    elif celeb == "Rahul Dravid":
+                        dravid = cv2.imread("rahul.jpg")
+                        st.image(dravid,width=150, caption='Rahul Dravid')
+                    elif celeb == "Sachin Tendulkar":
+                        sachin = cv2.imread("sachin.jpg")
+                        st.image(sachin,width=150, caption='Sachin Tendulkar')
+                    elif celeb == "Saurav Ganguly":
+                        ganguly = cv2.imread("ganguly.jpg")
+                        st.image(ganguly,width=150, caption='Saurav Ganguly')
+                    elif celeb == "virat_kohli":
+                        virat = cv2.imread("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')
+             except:
+                st.header("Image Cannot be Classified!")