Update app.py

app.py

@@ -20,17 +20,20 @@ bytes_data=None
 
 app = Flask(__name__)
 
-
+flag = True
 
 @app.route('/at')
 def testme():
     # return "i am in 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()
+    while flag :
+        
+        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:
+            flag = False
+            st.stop()
 
 # def attend():
 #     # Face recognition variables