loadimg from 🤗

setup.py

@@ -3,7 +3,7 @@
 
 setup(
     name="loadimg",
-    version="0.1.1",
+    version="0.1.2",
     description="a python package for loading images",
     long_description=pathlib.Path("README.md").read_text(encoding="utf-8"),
     long_description_content_type="text/markdown",

src/loadimg/utils.py

@@ -15,10 +15,12 @@ def download_image(url: str):
     try:
         if "github" in url and "raw=true" not in url:
             url += "?raw=true"
-        if "drive" in url and "uc?id=" not in url:
+        elif "drive" in url and "uc?id=" not in url:
             if "/view" in url or url.endswith("/"):
                 url = "/".join(url.split("/")[:-1])
             url = "https://drive.google.com/uc?id=" + url.split("/")[-1]
+        elif "hf.co" or "huggingface.co" in url : 
+            url = url.replace("/blob/","/resolve")
         response = requests.get(url, timeout=5)  # Timeout set to 5 seconds
         response.raise_for_status()  # Raise an exception for HTTP errors
         return Image.open(BytesIO(response.content))