Daniel Gatis 3 years ago
parent
commit
33e02fe732
1 changed files with 18 additions and 6 deletions
  1. 18 6
      rembg/cli.py

+ 18 - 6
rembg/cli.py

@@ -9,7 +9,7 @@ import click
 import filetype
 import uvicorn
 from asyncer import asyncify
-from fastapi import Depends, FastAPI, File, Query, Form
+from fastapi import Depends, FastAPI, File, Form, Query
 from fastapi.middleware.cors import CORSMiddleware
 from starlette.responses import Response
 from tqdm import tqdm
@@ -294,10 +294,16 @@ def s(port: int, log_level: str) -> None:
             ),
             a: bool = Query(default=False, description="Enable Alpha Matting"),
             af: int = Query(
-                default=240, ge=0, le=255, description="Alpha Matting (Foreground Threshold)"
+                default=240,
+                ge=0,
+                le=255,
+                description="Alpha Matting (Foreground Threshold)",
             ),
             ab: int = Query(
-                default=10, ge=0, le=255, description="Alpha Matting (Background Threshold)"
+                default=10,
+                ge=0,
+                le=255,
+                description="Alpha Matting (Background Threshold)",
             ),
             ae: int = Query(
                 default=10, ge=0, description="Alpha Matting (Erode Structure Size)"
@@ -320,10 +326,16 @@ def s(port: int, log_level: str) -> None:
             ),
             a: bool = Form(default=False, description="Enable Alpha Matting"),
             af: int = Form(
-                default=240, ge=0, le=255, description="Alpha Matting (Foreground Threshold)"
+                default=240,
+                ge=0,
+                le=255,
+                description="Alpha Matting (Foreground Threshold)",
             ),
             ab: int = Form(
-                default=10, ge=0, le=255, description="Alpha Matting (Background Threshold)"
+                default=10,
+                ge=0,
+                le=255,
+                description="Alpha Matting (Background Threshold)",
             ),
             ae: int = Form(
                 default=10, ge=0, description="Alpha Matting (Erode Structure Size)"
@@ -336,7 +348,7 @@ def s(port: int, log_level: str) -> None:
             self.ab = ab
             self.ae = ae
             self.om = om
-            
+
     def im_without_bg(content: bytes, commons: CommonQueryParams) -> Response:
         return Response(
             remove(