Update utility/utils.py

utility/utils.py

@@ -47,21 +47,13 @@ client = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.3", token=HFT)
 
 # Load image using OpenCV
 def load_image(image_path):
-    ext = os.path.splitext(file_path)[1].lower()
-    if ext in ['.png', '.jpg', '.jpeg']:
+    ext = os.path.splitext(image_path)[1].lower()
+    if ext in ['.png', '.jpg', '.jpeg', '.webp', '.tiff']:
         image = cv2.imread(image_path)
         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")
    
-def extract_hyperlinks(file_path):
-    ext = os.path.splitext(file_path)[1].lower()
-    
-    if ext in ['.png', '.jpg', '.jpeg']:
-        return 'Not Supported'
-    else:
-        raise ValueError("Unsupported file format")
-
 # Function for upscaling image using OpenCV's INTER_CUBIC
 def upscale_image(image, scale=2):
     height, width = image.shape[:2]