From 46b08240c19a9ee94807b941e0bc4e08cc1fb4bd Mon Sep 17 00:00:00 2001
From: Feidorian <brianmile10@yahoo.com>
Date: Fri, 24 Nov 2023 23:10:33 -0500
Subject: [PATCH] fixed is_valid_image utility function

---
 utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/utils.py b/utils.py
index 688fb24..6ea2067 100644
--- a/utils.py
+++ b/utils.py
@@ -5,6 +5,7 @@
 from contextlib import contextmanager
 import io
 import sys
+import os
 from datetime import datetime, timedelta
 import torch
 import numpy as np
@@ -55,7 +56,8 @@ def generate_mappings(mappings):
     }
 
 
-def is_valid_image(path: str) -> bool:
+def is_valid_image(dir: str, filename:str) -> bool:
+    path = os.path.join(dir, filename)
     try:
         img = Image.open(path)
         img.verify()