Upload api_only.py

iopaint/api_only.py

@@ -45,6 +45,9 @@ logger.info(f"Allowed origins: {allowed_origins}")
 # 初始化FastAPI
 app = FastAPI(title="IOPaint API")
 
+# 读取disable_nsfw环境变量，确保两个字段使用相同的值
+disable_nsfw_value = os.environ.get("IOPAINT_DISABLE_NSFW", "false").lower() == "true"
+
 # 配置API，添加所有缺失的必填字段
 config = ApiConfig(
     host=host,
@@ -57,14 +60,14 @@ config = ApiConfig(
     low_mem=os.environ.get("IOPAINT_LOW_MEM", "true").lower() == "true",
     no_half=os.environ.get("IOPAINT_NO_HALF", "false").lower() == "true",
     cpu_offload=os.environ.get("IOPAINT_CPU_OFFLOAD", "false").lower() == "true",
-    disable_nsfw=os.environ.get("IOPAINT_DISABLE_NSFW", "false").lower() == "true",
-    # 添加缺失的必填字段
+    disable_nsfw=disable_nsfw_value,
+    # 添加之前缺失的必填字段
     enable_interactive_seg=False,
     interactive_seg_model=InteractiveSegModel.sam2_1_tiny,
     interactive_seg_device=Device.cpu,
     enable_remove_bg=False,
     remove_bg_device=Device.cpu,
-    remove_bg_model=RemoveBGModel.briaai_rmbg_1_4,
+    remove_bg_model="briaai/RMBG-1.4",  # 修正为字符串类型而不是枚举
     enable_anime_seg=False,
     enable_realesrgan=False,
     realesrgan_device=Device.cpu,
@@ -73,6 +76,13 @@ config = ApiConfig(
     gfpgan_device=Device.cpu,
     enable_restoreformer=False,
     restoreformer_device=Device.cpu,
+    # 添加新发现缺失的必填字段
+    inbrowser=False,
+    disable_nsfw_checker=disable_nsfw_value,  # 使用相同的值
+    local_files_only=False,
+    cpu_textencoder=False,
+    mask_dir=None,
+    quality=100,
 )
 
 # 配置CORS