fix: get model module

DECIMER/__init__.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-"""DECIMER V2.4.0 Python Package. ============================
+"""DECIMER V2.6.0 Python Package. ============================
 
 This repository contains DECIMER-V2,Deep lEarning for Chemical ImagE Recognition) project
 was launched to address the OCSR problem with the latest computational intelligence methods
@@ -19,11 +19,11 @@
 please raise a issue on the Github repository.
 """
 
-__version__ = "2.4.0"
+__version__ = "2.6.0"
 
 __all__ = [
     "DECIMER",
 ]
 
 
-from .decimer import predict_SMILES, predict_SMILES_with_confidence
+from .decimer import predict_SMILES

DECIMER/decimer.py

@@ -36,7 +36,7 @@
 }
 
 
-def get_models(model_urls: dict) -> Tuple[object, tf.saved_model, tf.saved_model]:
+def get_models(model_urls: dict):
     """Download and load models from the provided URLs.
 
     This function downloads models from the provided URLs to a default location,

setup.py

@@ -8,7 +8,7 @@
 ) and platform.system() == "Darwin":
     tensorflow_os = "tensorflow-macos>=2.10.0"
 else:
-    tensorflow_os = "tensorflow==2.12.0"
+    tensorflow_os = "tensorflow>=2.12.0"
 
 with open("README.md", "r") as fh:
     long_description = fh.read()

tests/test_functions.py

@@ -16,7 +16,7 @@ def test_imagetosmilesWithConfidence():
     img_path = "tests/caffeine.png"
     actual_result = predict_SMILES(img_path, confidence=True)
 
-    for element, confidence in actual_result:
+    for element, confidence in actual_result[1]:
         assert (
             confidence >= 0.9
         ), f"Confidence for element '{element}' is below 0.9 (confidence: {confidence})"