Update app/utils.py

app/utils.py

@@ -4,14 +4,22 @@ from transformers import AutoModel, AutoTokenizer
 from PIL import Image, ImageEnhance, ImageFilter
 import torch
 import logging
-from huggingface_hub import delete_cache
+import shutil
+import os
 
 
 logger = logging.getLogger(__name__)
 
 class OCRModel:
     _instance = None
-
+    
+    def clear_huggingface_cache(self):
+            cache_dir = os.path.expanduser("~/.cache/huggingface")
+            if os.path.exists(cache_dir):
+                shutil.rmtree(cache_dir)
+                print("Hugging Face cache cleared.")
+            else:
+                print("No Hugging Face cache found.")
     def __new__(cls):
         if cls._instance is None:
             cls._instance = super(OCRModel, cls).__new__(cls)
@@ -21,7 +29,8 @@ class OCRModel:
     def initialize(self):
         try:
 
-            delete_cache()
+            self.clear_huggingface_cache()
+
             logger.info("Initializing OCR model...")
             # Try loading with use_fast=False if the fast tokenizer fails
             try: