Prateek954 commited on
Commit
456694e
1 Parent(s): ccafaf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ myList = os.listdir(directory)
18
 
19
  st.write("Photographs found in folder : ")
20
  for cls in myList:
21
- if os.path.splitext(cls)[1] in [".jpg", ".jpeg"]:
22
  img_path = os.path.join(directory, cls)
23
  curImg = cv2.imread(img_path)
24
  Images.append(curImg)
 
18
 
19
  st.write("Photographs found in folder : ")
20
  for cls in myList:
21
+ if os.path.splitext(cls)[1] in [".jpg", ".jpeg", "jfif"]:
22
  img_path = os.path.join(directory, cls)
23
  curImg = cv2.imread(img_path)
24
  Images.append(curImg)