diff --git a/__init__.py b/__init__.py index 84e315c..5a10a00 100644 --- a/__init__.py +++ b/__init__.py @@ -24,7 +24,7 @@ class SendToTostAIWebhook: def send_to_tost_ai_webhook(self, images, webhook_url: str, webhook_key: str,): (full_output_folder, filename, counter, subfolder, _,) = folder_paths.get_save_image_path("final_tost", folder_paths.get_temp_directory()) - tost_file_path = os.path.join(full_output_folder, f"{filename}_.png") + tost_file_path = os.path.join(full_output_folder, f"{filename}.png") tost_image = 255.0 * images[0].cpu().numpy() tost_image_pil = Image.fromarray(tost_image.astype(np.uint8)) tost_image_pil.save(tost_file_path)