Update utility/utils.py

utility/utils.py

@@ -45,15 +45,16 @@ HFT = os.getenv('HF_TOKEN')
 # Initialize the InferenceClient
 client = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.3", token=HFT)
 
-# Load image using OpenCV
 def load_image(image_path):
     ext = os.path.splitext(image_path)[1].lower()
     if ext in ['.png', '.jpg', '.jpeg', '.webp', '.tiff']:
         image = cv2.imread(image_path)
+        if image is None:
+            raise ValueError(f"Failed to load image from {image_path}. The file may be corrupted or unreadable.")
         return image
     else:
-        raise ValueError(f"Could not load image from {image_path}. It may be corrupted or the path is incorrect. or in a not supported format")
-   
+        raise ValueError(f"Unsupported image format: {ext}")   
+        
 # Function for upscaling image using OpenCV's INTER_CUBIC
 def upscale_image(image, scale=2):
     height, width = image.shape[:2]