Update app.py

app.py

@@ -17,7 +17,7 @@ myList = os.listdir(directory)
 
 st.write("Photographs found in folder : ")
 for cls in myList:
-    if os.path.splitext(cls)[1] in [".jpg", ".jpeg"]:
+    if os.path.splitext(cls)[1] in [".jpg", ".jpeg", ".png"]:
         img_path = os.path.join(directory, cls)
         curImg = cv2.imread(img_path)
         Images.append(curImg)