Merge pull request #96 from Kohulan/development

fix: bug with incorrect size of DECIMER_HandDrawn model

DECIMER/utils.py

@@ -76,19 +76,23 @@ def ensure_models(default_path: str, model_urls: dict) -> dict:
         dict: A dictionary containing model names as keys and their local paths as values
     """
     model_paths = {}
-
+    # Store st_size of each model
+    model_sizes = {
+        "DECIMER": 28080309,
+        "DECIMER_HandDrawn": 28080328,
+    }
     for model_name, model_url in model_urls.items():
         model_path = os.path.join(default_path, f"{model_name}_model")
         if (
-            os.path.exists(model_path)
-            and os.stat(os.path.join(model_path, "saved_model.pb")).st_size != 28080309
+                os.path.exists(model_path)
+                and os.stat(os.path.join(model_path, "saved_model.pb")).st_size != model_sizes.get(model_name)
         ):
+            print(f'Working with model {model_name}')
             shutil.rmtree(model_path)
             config.download_trained_weights(model_url, default_path)
         elif not os.path.exists(model_path):
             config.download_trained_weights(model_url, default_path)
 
         # Store the model path
         model_paths[model_name] = model_path
-
     return model_paths