Spaces:
Runtime error
Runtime error
Prateek954
commited on
Commit
•
456694e
1
Parent(s):
ccafaf1
Update app.py
Browse files
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)
|