update limits of remove bg, and update the assert messages (#36)

siliconflow · Jul 16, 2024 · b95e10a · b95e10a
1 parent 7723f40
commit b95e10a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/supernode.py b/supernode.py
@@ -78,7 +78,7 @@ def remove_background(self, image):
         device = image.device
         _, w, h, _ = image.shape
         assert (
-            w <= 1024 and h <= 1024
+            w <= 1536 and h <= 1536
         ), f"width and height must be less than 1024, but got {w} and {h}"
 
         payload = {

diff --git a/utils.py b/utils.py
@@ -115,7 +115,9 @@ def format_bytes(num_bytes: int) -> str:
 
 
 def validate_api_key(key: str):
-    assert isinstance(key, str), f"API_KEY must be a string, got {type(key)}"
+    assert isinstance(
+        key, str
+    ), f"API_KEY must be a string, got {type(key)}. Please refer to https://cloud.siliconflow.cn to get your API_KEY"
     assert key.startswith(
         "sk"
     ), f"invalid key, please refer to https://cloud.siliconflow.cn to get your API_KEY, got {key}"