mapreader 1.8.2.post0.dev9__tar.gz → 1.8.2.post0.dev15__tar.gz

{mapreader-1.8.2.post0.dev9 → mapreader-1.8.2.post0.dev15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mapreader
-Version: 1.8.2.post0.dev9
+Version: 1.8.2.post0.dev15
 Summary: A computer vision pipeline for the semantic exploration of maps/images at scale
 Home-page: https://github.com/Living-with-machines/MapReader
 Download-URL: https://github.com/Living-with-machines/MapReader/archive/refs/heads/main.zip

{mapreader-1.8.2.post0.dev9 → mapreader-1.8.2.post0.dev15}/mapreader/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2026-02-26T15:15:54+0000",
+ "date": "2026-03-25T21:06:12+0000",
  "dirty": false,
  "error": null,
- "full-revisionid": "87578a471ab1dcadf80a2552aa31a3f50cac862a",
- "version": "1.8.2.post0.dev9"
+ "full-revisionid": "08fceec9e3feef2a5c60385dba516f7639897f03",
+ "version": "1.8.2.post0.dev15"
 }
 '''  # END VERSION_JSON
 

{mapreader-1.8.2.post0.dev9 → mapreader-1.8.2.post0.dev15}/mapreader/classify/classifier.py

@@ -108,6 +108,7 @@ class ClassifierContainer:
         is_inception: bool = False,
         load_path: str | None = None,
         force_device: bool = False,
+        huggingface: bool = False,
         **kwargs,
     ):
         # set up device
@@ -149,7 +150,31 @@ class ClassifierContainer:
                 self.input_size = input_size
                 self.is_inception = is_inception
             elif isinstance(model, str):
-                self._initialize_model(model, **kwargs)
+                if huggingface:
+                    try:
+                        from transformers import AutoModelForImageClassification, AutoImageProcessor
+                    except ImportError:
+                        raise ImportError(
+                            "Hugging Face models require the 'transformers' library: 'pip install transformers'."
+                        )
+                    print(f"[INFO] Initializing Hugging Face model: {model}")
+                    num_labels = len(self.labels_map)
+                    self.model = AutoModelForImageClassification.from_pretrained(
+                        model,
+                        num_labels=num_labels,
+                        ignore_mismatched_sizes=True
+                    ).to(self.device)
+                    hf_processor = AutoImageProcessor.from_pretrained(model)
+                    size = getattr(hf_processor, "size", {})
+                    if "height" in size and "width" in size:
+                        size = (size["height"], size["width"])
+                    elif "shortest_edge" in size:
+                        size = (size["shortest_edge"], size["shortest_edge"])
+                    else:
+                        size = input_size
+                    self.is_inception = False
+                else:
+                    self._initialize_model(model, **kwargs)
 
             self.optimizer = None
             self.scheduler = None

{mapreader-1.8.2.post0.dev9 → mapreader-1.8.2.post0.dev15}/mapreader.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mapreader
-Version: 1.8.2.post0.dev9
+Version: 1.8.2.post0.dev15
 Summary: A computer vision pipeline for the semantic exploration of maps/images at scale
 Home-page: https://github.com/Living-with-machines/MapReader
 Download-URL: https://github.com/Living-with-machines/MapReader/archive/refs/heads/main.zip