add remove_bg function

2023-06-20 02:45:37 -07:00
parent c7d3c047fd
commit baca747bbb
9 changed files with 94 additions and 2 deletions
@@ -260,6 +260,7 @@ NOTE: Just reply using these information, don't ask me anything.
        logger.debug("Start calling stable_diffusion")
        img = await call_sd(sd_params)
        logger.debug("End calling stable_diffusion")
+        context.set_last_image(img)
        await context.reply_image_base64(img)
        return "Success"