Update app.py

app.py

@@ -107,7 +107,7 @@ def generate_charts(ner_output_ext: dict) -> Tuple[go.Figure, np.ndarray]:
     # Generate word cloud
     wordcloud_image = generate_wordcloud(ner_output_ext['entities'], ext_color_map, "dh3.png")
     
-    return fig2, wordcloud_image
+    return fig1, wordcloud_image
 
 def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str], file_path: str) -> np.ndarray:
     # Construct the absolute path