Update app.py

app.py

@@ -1,3 +1,4 @@
+####### Section 1 ###################
 from PIL import Image
 import numpy as np
 import cv2
@@ -6,8 +7,10 @@ import face_recognition
 import os
 import streamlit as st
 
+####### Section 2 ###################
 p1 = "sarwan.jpg"
 p2 = "rattantata.png"
+p3 = "Ravinder.jpg"
 
 st.title("Face Recognition ")
 Images     = []
@@ -22,12 +25,17 @@ img2 = cv2.imread(p2)
 Images.append(img2)
 classnames.append("Rattan Tata")
 
+img3 = cv2.imread(p3)
+Images.append(img3)
+classnames.append("Ravinder kaur")
+
 # Load images for face recognition
 encodeListknown = [face_recognition.face_encodings(img)[0] for img in Images]
 
-# take  image from user  
+
+####### Section 3 ################### 
 # Take picture using the camera 
-img_file_buffer = st.camera_input("Take a picture")
+img_file_buffer = st.camera_input("Take Your picture")
 
 # recognise the face in the uploaded image 
 if img_file_buffer is not None:
@@ -49,7 +57,7 @@ if img_file_buffer is not None:
 
         if matches[matchIndex]:
             name = classnames[matchIndex].upper()
-            st.write (name)
+            #st.write (name)
             # show the name on image to user 
             y1, x2, y2, x1 = faceLoc
             y1, x2, y2, x1 = y1 * 4, x2 * 4, y2 * 4, x1 * 4
@@ -59,8 +67,7 @@ if img_file_buffer is not None:
 
             faceMatchedflag = 1
      
-    st.image(image , use_column_width=True, output_format="PNG")
-            
+    st.image(image , use_column_width=True, output_format="PNG")       
             
     if(faceMatchedflag == 0) : 
         st.warning("No faces detected in the image.")