Update app.py

app.py

@@ -14,7 +14,7 @@ pylab.rcParams['figure.figsize'] = (10.0, 8.0) # this controls figure size in th
 
 import io
 import streamlit as st
-
+bytes_data=None
 
 ##################################################3
 
@@ -23,6 +23,14 @@ app = Flask(__name__)
 
 
 @app.route('/at')
+def testme():
+    img_file_buffer=st.camera_input("Take a picture")
+    if img_file_buffer is not None:
+         test_image = Image.open(img_file_buffer)
+         st.image(test_image, use_column_width=True)
+    if bytes_data is None:
+     st.stop()
+
 def attend():
     # Face recognition variables
     known_faces_names = ["Sarwan Sir", "Vikas","Lalit","Jasmeen","Anita Ma'am"]