Spaces:
Sleeping
Sleeping
MonkeyJuice
commited on
Commit
•
6bd4266
1
Parent(s):
6266d6b
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ def predict_batch(zip_file, score_threshold: float, progress=gr.Progress()):
|
|
26 |
with zipfile.ZipFile(zip_file) as zf:
|
27 |
for file in progress.tqdm(zf.namelist()):
|
28 |
print(file)
|
29 |
-
if file.endswith(".png") or file.endswith(".jpg"):
|
30 |
image_file = zf.open(file)
|
31 |
image = PIL.Image.open(image_file)
|
32 |
image = image.convert("RGB")
|
|
|
26 |
with zipfile.ZipFile(zip_file) as zf:
|
27 |
for file in progress.tqdm(zf.namelist()):
|
28 |
print(file)
|
29 |
+
if file.endswith(".png") or file.endswith(".jpg") or file.endswith(".jpeg"):
|
30 |
image_file = zf.open(file)
|
31 |
image = PIL.Image.open(image_file)
|
32 |
image = image.convert("RGB")
|