add realesrGAN selection

2024-02-08 17:16:57 +08:00
parent 8060e16c70
commit f52dbc1091
6 changed files with 98 additions and 9 deletions
--- a/iopaint/web_config.py
+++ b/iopaint/web_config.py
@@ -44,7 +44,7 @@ default_configs = dict(
    interactive_seg_model=InteractiveSegModel.vit_b,
    interactive_seg_device=Device.cpu,
    enable_remove_bg=False,
-    remove_bg_model=RemoveBGModel.u2net,
+    remove_bg_model=RemoveBGModel.briaai_rmbg_1_4,
    enable_anime_seg=False,
    enable_realesrgan=False,
    realesrgan_device=Device.cpu,