From e7c33fff47617114f78266f58888b555b8b6914d Mon Sep 17 00:00:00 2001
From: daxcay <daxtoncaylor@gmail.com>
Date: Fri, 5 Jul 2024 11:27:17 +0530
Subject: [PATCH] Update DATASET_SaveImage.py

---
 classes/DATASET_SaveImage.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/classes/DATASET_SaveImage.py b/classes/DATASET_SaveImage.py
index 2ac9720..91047cb 100644
--- a/classes/DATASET_SaveImage.py
+++ b/classes/DATASET_SaveImage.py
@@ -1,7 +1,8 @@
 import os
 import json
 import numpy as np
-from PIL import Image, PngInfo
+from PIL import Image
+from PIL.PngImagePlugin import PngInfo
 from comfy.cli_args import args
 
 class DataSet_SaveImage: