fix #434

danielgatis · Apr 27, 2023 · f2353d0 · f2353d0
1 parent e47b2a0
commit f2353d0
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/rembg/commands/s_command.py b/rembg/commands/s_command.py
@@ -1,5 +1,5 @@
 import json
-from typing import Annotated, Optional, Tuple, cast
+from typing import Optional, Tuple, cast
 
 import aiohttp
 import click
@@ -83,10 +83,10 @@ def s_command(port: int, log_level: str, threads: int) -> None:
     class CommonQueryParams:
         def __init__(
             self,
-            model: Annotated[
-                str, Query(regex=r"(" + "|".join(sessions_names) + ")")
-            ] = Query(
+            model: str = Query(
                 description="Model to use when processing image",
+                regex=r"(" + "|".join(sessions_names) + ")",
+                default="u2net",
             ),
             a: bool = Query(default=False, description="Enable Alpha Matting"),
             af: int = Query(
@@ -128,10 +128,10 @@ def __init__(
     class CommonQueryPostParams:
         def __init__(
             self,
-            model: Annotated[
-                str, Form(regex=r"(" + "|".join(sessions_names) + ")")
-            ] = Form(
+            model: str = Form(
                 description="Model to use when processing image",
+                regex=r"(" + "|".join(sessions_names) + ")",
+                default="u2net",
             ),
             a: bool = Form(default=False, description="Enable Alpha Matting"),
             af: int = Form(