Upload api_only.py

iopaint/api_only.py

@@ -14,7 +14,8 @@ from iopaint.const import DEFAULT_MODEL_DIR
 # 从环境变量读取配置
 host = os.environ.get("IOPAINT_HOST", "0.0.0.0")
 port = int(os.environ.get("IOPAINT_PORT", "7860"))
-model = os.environ.get("IOPAINT_MODEL", "lama")
+# 修改默认模型为cv2，因为lama无法加载
+model = os.environ.get("IOPAINT_MODEL", "cv2")
 
 # 修改模型目录路径，使用/app或/tmp目录
 model_dir_str = os.environ.get("IOPAINT_MODEL_DIR", "/app/models")
@@ -83,6 +84,9 @@ config = ApiConfig(
     cpu_textencoder=False,
     mask_dir=None,
     quality=100,
+    # 添加cv2模型需要的特定参数
+    cv2_radius=5,
+    cv2_flag="INPAINT_NS",
 )
 
 # 配置CORS