(self, inputs)
| 595 | "The input to this tool should be a comma separated string of two, " |
| 596 | "representing the image_path and the user description") |
| 597 | def inference(self, inputs): |
| 598 | image_path, instruct_text = inputs.split(",")[0], ','.join(inputs.split(',')[1:]) |
| 599 | image = Image.open(image_path) |
| 600 | self.seed = random.randint(0, 65535) |
| 601 | seed_everything(self.seed) |
| 602 | prompt = f'{instruct_text}, {self.a_prompt}' |
| 603 | image = self.pipe(prompt, image, num_inference_steps=20, eta=0.0, negative_prompt=self.n_prompt, |
| 604 | guidance_scale=9.0).images[0] |
| 605 | updated_image_path = get_new_image_name(image_path, func_name="pose2image") |
| 606 | image.save(updated_image_path) |
| 607 | print(f"\nProcessed PoseText2Image, Input Pose: {image_path}, Input Text: {instruct_text}, " |
| 608 | f"Output Image: {updated_image_path}") |
| 609 | return updated_image_path |
| 610 | |
| 611 | class SegText2Image: |
| 612 | def __init__(self, device): |
nothing calls this directly
no test coverage detected