Update mdr_pdf_parser.py

mdr_pdf_parser.py

@@ -47,7 +47,8 @@ from munch import Munch
 from transformers import LayoutLMv3ForTokenClassification
 import onnxruntime
 # --- HUGGING FACE HUB IMPORT ONLY BECAUSE RUNNING IN SPACES NOT NECESSARY IN PROD ---
-from huggingface_hub import hf_hub_download, HfHubDownloadError
+from huggingface_hub import hf_hub_download
+from huggingface_hub.errors import HfHubHTTPError
 import time # Added for example usage timing
 
 # --- External Dependencies ---
@@ -1890,14 +1891,16 @@ class MDRExtractionEngine:
             self._yolo = YOLOv10(yolo_model_filepath)
             print("MDR YOLOv10 model loaded successfully.")
 
-        except HfHubDownloadError as e:
-            print(f"ERROR: Failed to download YOLO model from Hugging Face Hub: {e}")
+        # --- MODIFIED EXCEPTION HANDLING ---
+        except HfHubHTTPError as e: # <-- CHANGED THIS LINE
+            print(f"ERROR: Failed to download/access YOLO model via Hugging Face Hub: {e}") # Slightly updated message
             self._yolo = None
-        except FileNotFoundError as e: # Catch if hf_hub_download fails finding file
-             print(f"ERROR: YOLO model file not found via Hugging Face Hub: {e}")
+        except FileNotFoundError as e: # Catch if hf_hub_download fails finding file OR YOLOv10 constructor fails
+             print(f"ERROR: YOLO model file not found or failed to load locally: {e}") # Slightly updated message
              self._yolo = None
         except Exception as e:
-            print(f"ERROR: Failed to load YOLOv10 model from {yolo_model_filepath}: {e}")
+            # Keep the general exception catch, but make the message more specific
+            print(f"ERROR: An unexpected issue occurred loading YOLOv10 model from {yolo_cache_dir}/{filename}: {e}")
             self._yolo = None
 
     elif YOLOv10 is None: