Update working_yolo_pipeline.py

working_yolo_pipeline.py

@@ -3,6 +3,7 @@ import fitz  # PyMuPDF
 import numpy as np
 import cv2
 import torch
+import torch
 import torch.serialization
 
 _original_torch_load = torch.load
@@ -2656,15 +2657,15 @@ def run_document_pipeline(input_pdf_path: str, layoutlmv3_model_path: str, struc
         traceback.print_exc()
         return None
 
-    finally:
-        print(f"\nCleaning up temporary files in {temp_pipeline_dir}...")
-        try:
-            for f in glob.glob(os.path.join(temp_pipeline_dir, '*')):
-                os.remove(f)
-            os.rmdir(temp_pipeline_dir)
-            print("🧹 Cleanup successful.")
-        except Exception as e:
-            print(f"⚠️ Cleanup failed: {e}")
+    # finally:
+    #     print(f"\nCleaning up temporary files in {temp_pipeline_dir}...")
+    #     try:
+    #         for f in glob.glob(os.path.join(temp_pipeline_dir, '*')):
+    #             os.remove(f)
+    #         os.rmdir(temp_pipeline_dir)
+    #         print("🧹 Cleanup successful.")
+    #     except Exception as e:
+    #         print(f"⚠️ Cleanup failed: {e}")
 
     total_time = time.time() - overall_start
     print("\n" + "#" * 80)