AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-05 16:40:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #434

Browse Source
This commit is contained in:
Daniel Gatis 2023-04-27 20:01:37 -03:00
parent e47b2a0ed4
commit f2353d09e2
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
rembg/commands/s_command.py
View File

@ -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 s_command(port: int, log_level: str, threads: int) -> None:
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(