Update app.py

app.py

@@ -104,12 +104,10 @@ def generate_charts(ner_output_ext: dict) -> Tuple[go.Figure, np.ndarray]:
         plot_bgcolor='rgba(0,0,0,0)',
         paper_bgcolor='rgba(0,0,0,0)'
     )
-    
-    
     # Generate word cloud
-    wordcloud_image = generate_wordcloud(ner_output_ext['entities'], ext_color_map, "dh3.img")
+    wordcloud_image = generate_wordcloud(ner_output_bin['entities'], bin_color_map, "dh3.png")
     
-    return fig1, wordcloud_image
+    return fig2, wordcloud_image
 
 def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str], file_path: str) -> np.ndarray:
     # Construct the absolute path