MonkeyJuice commited on
Commit
6bd4266
1 Parent(s): 6266d6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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")