useknockout 0.0.3__tar.gz → 0.0.4__tar.gz

{useknockout-0.0.3 → useknockout-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: useknockout
-Version: 0.0.3
+Version: 0.0.4
 Summary: Python SDK for useknockout — state-of-the-art background removal API.
 Project-URL: Homepage, https://useknockout.com
 Project-URL: Documentation, https://github.com/useknockout/python

{useknockout-0.0.3 → useknockout-0.0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "useknockout"
-version = "0.0.3"
+version = "0.0.4"
 description = "Python SDK for useknockout — state-of-the-art background removal API."
 readme = "README.md"
 license = { text = "MIT" }

useknockout-0.0.4/src/useknockout/_version.py

@@ -0,0 +1 @@
+__version__ = "0.0.4"

{useknockout-0.0.3 → useknockout-0.0.4}/src/useknockout/async_client.py

@@ -329,16 +329,24 @@ class AsyncKnockout:
         file: FileInput,
         *,
         scale: int = 4,
+        model: str = "swin2sr",
         face_enhance: bool = False,
         format: str = "png",
     ) -> bytes:
         if scale not in (2, 4):
             raise ValueError("scale must be 2 or 4")
+        if model not in ("swin2sr", "realesrgan"):
+            raise ValueError("model must be 'swin2sr' or 'realesrgan'")
         return await self._request_bytes(
             "POST",
             "/upscale",
             files=_multipart_files(file),
-            data=_form({"scale": scale, "face_enhance": face_enhance, "format": format}),
+            data=_form({
+                "scale": scale,
+                "model": model,
+                "face_enhance": face_enhance,
+                "format": format,
+            }),
         )
 
     async def face_restore(

{useknockout-0.0.3 → useknockout-0.0.4}/src/useknockout/client.py

@@ -358,21 +358,30 @@ class Knockout:
         file: FileInput,
         *,
         scale: int = 4,
+        model: str = "swin2sr",
         face_enhance: bool = False,
         format: str = "png",
     ) -> bytes:
-        """POST /upscale — Real-ESRGAN x2/x4 super-resolution.
+        """POST /upscale — 2x/4x super-resolution.
 
-        ``face_enhance=True`` routes through GFPGAN for sharper facial detail
-        (slower; use for portraits).
+        ``model="swin2sr"`` (default, v0.6.0+) is sharper on real photos.
+        ``model="realesrgan"`` is the legacy backend — better on anime / illustrations.
+        ``face_enhance=True`` routes through GFPGAN (Real-ESRGAN backend).
         """
         if scale not in (2, 4):
             raise ValueError("scale must be 2 or 4")
+        if model not in ("swin2sr", "realesrgan"):
+            raise ValueError("model must be 'swin2sr' or 'realesrgan'")
         return self._request_bytes(
             "POST",
             "/upscale",
             files=_multipart_files(file),
-            data=_form({"scale": scale, "face_enhance": face_enhance, "format": format}),
+            data=_form({
+                "scale": scale,
+                "model": model,
+                "face_enhance": face_enhance,
+                "format": format,
+            }),
         )
 
     def face_restore(

useknockout-0.0.3/src/useknockout/_version.py

@@ -1 +0,0 @@
-__version__ = "0.0.3"