fix bar plot

app.py

@@ -38,11 +38,11 @@ def greet(name, descriptions):
     # run keyword extraction
     kw_model = KeyBERT()
     keyword_extract(descriptions, kw_model, 1, save_output=f'results/{name}/{1}_keyword.png')
-    keyword_extract(descriptions, kw_model, 2, save_output=f'results/{name}/{2}_keyword.png')
-    keyword_extract(descriptions, kw_model, 3, save_output=f'results/{name}/{3}_keyword.png')
+    # keyword_extract(descriptions, kw_model, 2, save_output=f'results/{name}/{2}_keyword.png')
+    # keyword_extract(descriptions, kw_model, 3, save_output=f'results/{name}/{3}_keyword.png')
     outputs.append(Image.open(f'results/{name}/1_keyword.png'))
-    outputs.append(Image.open(f'results/{name}/2_keyword.png'))
-    outputs.append(Image.open(f'results/{name}/3_keyword.png'))
+    # outputs.append(Image.open(f'results/{name}/2_keyword.png'))
+    # outputs.append(Image.open(f'results/{name}/3_keyword.png'))
 
     # keywords = kw_model.extract_keywords(descriptions, highlight=True)
     # print(keywords)

main.py

@@ -72,8 +72,8 @@ filename.open('w', encoding='utf-8').write(svg)
 # run keyword extraction
 kw_model = KeyBERT()
 keyword_extract(doc, kw_model, 1, save_output=f'results/DAD/{1}_keyword.png')
-keyword_extract(doc, kw_model, 2, save_output=f'results/DAD/{2}_keyword.png')
-keyword_extract(doc, kw_model, 3, save_output=f'results/DAD/{3}_keyword.png')
+# keyword_extract(doc, kw_model, 2, save_output=f'results/DAD/{2}_keyword.png')
+# keyword_extract(doc, kw_model, 3, save_output=f'results/DAD/{3}_keyword.png')
 keywords = kw_model.extract_keywords(doc, highlight=True)
 print(keywords)