diff --git a/playground/augment-image.png b/playground/augment-image.png index 72019b5..d9c066d 100644 Binary files a/playground/augment-image.png and b/playground/augment-image.png differ diff --git a/playground/augment-image.py b/playground/augment-image.py index 273ef24..32f3993 100644 --- a/playground/augment-image.py +++ b/playground/augment-image.py @@ -32,7 +32,7 @@ async def generate( agent = AugmentImageInfer.build( req_type=request_type, image=image, - mood=Moods.Shy, + # mood=Moods.Shy, prompt="", defry=0, ) @@ -45,7 +45,7 @@ async def generate( print(f"Error: {e.message}") return None else: - print(f"Meta: {result.meta}") + print(f"Meta: {result.meta.endpoint}") _res: ImageGenerateResp file = result.files[0] with open(f"{pathlib.Path(__file__).stem}.png", "wb") as f: diff --git a/playground/generate_image.py b/playground/generate_image.py index c280871..a09eb9e 100755 --- a/playground/generate_image.py +++ b/playground/generate_image.py @@ -91,7 +91,7 @@ async def direct_use(): that's pydantic! :return: """ - credential = JwtCredential(jwt_token=SecretStr("pst-5555")) + credential = ApiCredential(api_token=SecretStr("pst-5555")) result = await GenerateImageInfer( input="1girl", model=Model.NAI_DIFFUSION_4_CURATED_PREVIEW,