From 3de1407d1811d1c3135a3f353ef3260947ab3a93 Mon Sep 17 00:00:00 2001
From: Robert Sachunsky <sachunsky@informatik.uni-leipzig.de>
Date: Thu, 21 May 2026 02:38:20 +0200
Subject: [PATCH] drop unnecessary TF / Torch imports

---
 src/eynollah/cli/__init__.py        |  4 ----
 src/eynollah/extract_images.py      |  7 -------
 src/eynollah/eynollah_imports.py    | 13 -------------
 src/eynollah/eynollah_ocr.py        |  4 ----
 src/eynollah/mb_ro_on_layout.py     |  4 ----
 src/eynollah/model_zoo/model_zoo.py |  4 ++++
 src/eynollah/ocrd_cli.py            |  6 ++----
 7 files changed, 6 insertions(+), 36 deletions(-)
 delete mode 100644 src/eynollah/eynollah_imports.py

diff --git a/src/eynollah/cli/__init__.py b/src/eynollah/cli/__init__.py
index 43ed046..1584fa5 100644
--- a/src/eynollah/cli/__init__.py
+++ b/src/eynollah/cli/__init__.py
@@ -1,7 +1,3 @@
-# NOTE: For predictable order of imports of torch/shapely/tensorflow
-#       this must be the first import of the CLI!
-from ..eynollah_imports import imported_libs
-
 from .cli import main
 from .cli_binarize import binarize_cli
 from .cli_enhance import enhance_cli
diff --git a/src/eynollah/extract_images.py b/src/eynollah/extract_images.py
index 7a7e3f6..40476a3 100644
--- a/src/eynollah/extract_images.py
+++ b/src/eynollah/extract_images.py
@@ -9,7 +9,6 @@ import os
 import time
 from typing import Optional
 from pathlib import Path
-import tensorflow as tf
 import numpy as np
 import cv2
 
@@ -64,12 +63,6 @@ class EynollahImageExtractor(Eynollah):
 
         t_start = time.time()
 
-        try:
-            for device in tf.config.list_physical_devices('GPU'):
-                tf.config.experimental.set_memory_growth(device, True)
-        except:
-            self.logger.warning("no GPU device available")
-            
         self.logger.info("Loading models...")
         self.setup_models()
         self.logger.info(f"Model initialization complete ({time.time() - t_start:.1f}s)")
diff --git a/src/eynollah/eynollah_imports.py b/src/eynollah/eynollah_imports.py
deleted file mode 100644
index 496406c..0000000
--- a/src/eynollah/eynollah_imports.py
+++ /dev/null
@@ -1,13 +0,0 @@
-"""
-Load libraries with possible race conditions once. This must be imported as the first module of eynollah.
-"""
-import os
-os.environ['TF_USE_LEGACY_KERAS'] = '1' # avoid Keras 3 after TF 2.15
-
-from ocrd_utils import tf_disable_interactive_logs
-from torch import *
-tf_disable_interactive_logs()
-import tensorflow.keras
-from shapely import *
-imported_libs = True
-__all__ = ['imported_libs']
diff --git a/src/eynollah/eynollah_ocr.py b/src/eynollah/eynollah_ocr.py
index 4470671..77ad98f 100644
--- a/src/eynollah/eynollah_ocr.py
+++ b/src/eynollah/eynollah_ocr.py
@@ -14,10 +14,6 @@ from cv2.typing import MatLike
 from xml.etree import ElementTree as ET
 from PIL import Image, ImageDraw
 import numpy as np
-try:
-    import torch
-except ImportError:
-    torch = None
 
 
 from .eynollah import Eynollah
diff --git a/src/eynollah/mb_ro_on_layout.py b/src/eynollah/mb_ro_on_layout.py
index 5725ba1..6c0477b 100644
--- a/src/eynollah/mb_ro_on_layout.py
+++ b/src/eynollah/mb_ro_on_layout.py
@@ -17,10 +17,6 @@ import cv2
 import numpy as np
 import statistics
 
-os.environ['TF_USE_LEGACY_KERAS'] = '1' # avoid Keras 3 after TF 2.15
-import tensorflow as tf
-from tensorflow.keras.models import Model
-
 from .eynollah import Eynollah
 from .model_zoo import EynollahModelZoo
 from .utils.resize import resize_image
diff --git a/src/eynollah/model_zoo/model_zoo.py b/src/eynollah/model_zoo/model_zoo.py
index 815663e..ec35a80 100644
--- a/src/eynollah/model_zoo/model_zoo.py
+++ b/src/eynollah/model_zoo/model_zoo.py
@@ -269,6 +269,10 @@ class EynollahModelZoo:
         """
         Load decoder for OCR
         """
+        os.environ['TF_USE_LEGACY_KERAS'] = '1' # avoid Keras 3 after TF 2.15
+        from ocrd_utils import tf_disable_interactive_logs
+        tf_disable_interactive_logs()
+
         from tensorflow.keras.layers import StringLookup
 
         characters = self._load_characters()
diff --git a/src/eynollah/ocrd_cli.py b/src/eynollah/ocrd_cli.py
index acd8d4e..effecb2 100644
--- a/src/eynollah/ocrd_cli.py
+++ b/src/eynollah/ocrd_cli.py
@@ -1,10 +1,8 @@
-# NOTE: For predictable order of imports of torch/shapely/tensorflow
-#       this must be the first import of the CLI!
-from .eynollah_imports import imported_libs
-from .processor import EynollahProcessor
 from click import command
 from ocrd.decorators import ocrd_cli_options, ocrd_cli_wrap_processor
 
+from .processor import EynollahProcessor
+
 @command()
 @ocrd_cli_options
 def main(*args, **kwargs):