Update app.py

app.py

@@ -109,7 +109,6 @@ def generate_charts(ner_output_ext: dict) -> Tuple[go.Figure, np.ndarray]:
     
     return fig1, wordcloud_image
 
-
 def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str]) -> np.ndarray:
     token_texts = []
     token_scores = []
@@ -117,9 +116,11 @@ def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str]) -> np.nd
 
     for entity in entities:
         for token in entity['tokens']:
-            token_texts.append(token)
+            cleaned_token = token.lstrip('_')
+            token_texts.append(cleaned_token)
             token_scores.append(entity['score'])
             token_types.append(entity['entity'])
+            print(f"{cleaned_token} ({entity['entity']}): {entity['score']}")
 
     # Create a dictionary for word cloud
     word_freq = {text: score for text, score in zip(token_texts, token_scores)}