nodes/daxcay-ComfyUI-DataSet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updates

Browse Source
This commit is contained in:
daxcay
2024-06-14 19:10:45 +05:30
parent 82ef606c4b
commit faf7ed9a07
2 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
classes/DATASET_CaptionVisualizer.py
+3 -3
View File
@@ -176,15 +176,15 @@ class DATASET_CaptionVisualizer:
OUTPUT_NODE = True
CATEGORY = "🔶DATASET🔶"
def Visualize(self, TextFilePathList, WordCloudTop, NetworkGraphTop, FrequencyGraphTop):
def Visualize(self, Captions, WordCloudTop, NetworkGraphTop, FrequencyGraphTop):
try:
directory_path = os.path.dirname(TextFilePathList[0])
directory_path = os.path.dirname(Captions[0])
visualize_path = os.path.join(directory_path, "visualize")
os.makedirs(visualize_path, exist_ok=True)
wc, ng, fg = generate_wordcloud_and_network_graph(
TextFilePathList, visualize_path, WordCloudTop[0], NetworkGraphTop[0], FrequencyGraphTop[0])
Captions, visualize_path, WordCloudTop[0], NetworkGraphTop[0], FrequencyGraphTop[0])
images = []
if os.path.exists(wc):