Add compatibility function for huggingface_hub download

DummyModel.py

@@ -4,13 +4,19 @@ import torch.nn
 from huggingface_hub import hf_hub_download
 
 
+def _hf_hub_download_compat(repo_id: str, filename: str, token: str) -> str:
+    try:
+        return hf_hub_download(repo_id, filename, token=token)
+    except TypeError:
+        # Backward compatibility for older huggingface_hub releases.
+        return hf_hub_download(repo_id, filename, use_auth_token=token)
+
+
 def load_dummy_model(DEBUG):
     model = DummyModel()
     if not DEBUG:
-        file_path = hf_hub_download(
-            "lfolle/DeepNAPSIModel",
-            "dummy_model.pth",
-            use_auth_token=os.environ["DeepNAPSIModel"],
+        file_path = _hf_hub_download_compat(
+            "lfolle/DeepNAPSIModel", "dummy_model.pth", os.environ["DeepNAPSIModel"]
         )
         model.load_state_dict(torch.load(file_path))
     return model

Model.py

@@ -3,6 +3,14 @@ from huggingface_hub import hf_hub_download
 from nail_classification.inference import Inference
 
 
+def _hf_hub_download_compat(repo_id: str, filename: str, token: str) -> str:
+    try:
+        return hf_hub_download(repo_id, filename, token=token)
+    except TypeError:
+        # Backward compatibility for older huggingface_hub releases.
+        return hf_hub_download(repo_id, filename, use_auth_token=token)
+
+
 class Model:
     def __init__(self, DEBUG):
         if DEBUG:
@@ -10,10 +18,8 @@ class Model:
             file_paths = [os.path.join(base, f"version_{v}") for v in range(10, 15)]
         else:
             file_paths = [
-                hf_hub_download(
-                    "lfolle/DeepNAPSIModel",
-                    f"version_{v}.ckpt",
-                    use_auth_token=os.environ["DeepNAPSIModel"],
+                _hf_hub_download_compat(
+                    "lfolle/DeepNAPSIModel", f"version_{v}.ckpt", os.environ["DeepNAPSIModel"]
                 )
                 for v in [10, 11, 12, 13, 14]
             ]