From 8d78632dd084e8fb4c078f9368e38f09069249fa Mon Sep 17 00:00:00 2001 From: stellahsr Date: Wed, 10 Apr 2024 16:33:38 +0800 Subject: [PATCH] update local code --- examples/di/rm_image_background.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/examples/di/rm_image_background.py b/examples/di/rm_image_background.py index cb7900a0a..524ecdb33 100644 --- a/examples/di/rm_image_background.py +++ b/examples/di/rm_image_background.py @@ -1,15 +1,16 @@ import asyncio -from metagpt.roles.di.data_interpreter import DataInterpreter +from metagpt.roles.di.mgx import MGX async def main(requirement: str = ""): - di = DataInterpreter() + # di = DataInterpreter() + di = MGX(use_intent=False, tools=[""]) await di.run(requirement) if __name__ == "__main__": - image_path = "/your/path/to/the/image.jpeg" - save_path = "/your/intended/save/path/for/image_rm_bg.png" + image_path = r"F:\deepWisdom\metaGPT\hsr\MetaGPT\examples\data\dog.beebf16d.jpg" + save_path = r"F:\deepWisdom\metaGPT\hsr\MetaGPT\examples\data\/image_rm_bg.png" requirement = f"This is a image, you need to use python toolkit rembg to remove the background of the image and save the result. image path:{image_path}; save path:{save_path}." asyncio.run(main(requirement))