Update app.py

app.py

@@ -1,29 +1,28 @@
-import streamlit as st
 from keras.models import load_model
 from PIL import Image
 import numpy as np
-import cv2   #module in openCV to provide easy to use interface for workin image and video
+import cv2 
 import requests
 
-import io  #for input output or simply read and write files operations
-bytes_data = None  # this under io, bytesio is classfor operation on byte data
-
 import face_recognition
-import os       #os module allows us to operate on underlying OS irrespective mac windows etc.
+import os
 from datetime import datetime
 
-from matplotlib import pyplot as plt      #matplotlib is package  lets you draw inline pictures in the notebooks
-import pylab                              #module that bulk imports matplotlib.pyplot (for plotting) and NumPy (for Mathematics and working with arrays) in a single name space.
-pylab.rcParams['figure.figsize'] = (10.0, 8.0) # this controls figure size 
-
+#the following are to do with this interactive notebook code
+from matplotlib import pyplot as plt # this lets you draw inline pictures in the notebooks
+import pylab # this allows you to control figure size 
+pylab.rcParams['figure.figsize'] = (10.0, 8.0) # this controls figure size in the notebook
 
+import io
+import streamlit as st
+bytes_data=None
 
 Images = []
 classnames = []
 myList = os.listdir()
 #st.write(myList)
 for cls in myList:
-    if os.path.splitext(cls)[1] == ".jpeg"  :
+    if os.path.splitext(cls)[1] == ".jpg"  :
         curImg = cv2.imread(f'{cls}')
         Images.append(curImg)
         classnames.append(os.path.splitext(cls)[0])
@@ -84,4 +83,4 @@ if img_file_buffer is not None:
      ##############################
      st.image(image)
 if bytes_data is None:
-     st.stop()
+     st.stop()