Update app.py

app.py

@@ -107,9 +107,11 @@ def generate_charts(ner_output_bin: dict) -> Tuple[go.Figure, np.ndarray]:
     
     return fig2, wordcloud_image
 
-def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str], image_path: str) -> np.ndarray:
+def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str], file_path: str) -> np.ndarray:
 
-    image_path = os.path.join(os.path.dirname(__file__), image_path)
+    image_path = os.path.join(os.path.dirname(__file__), file_path)
+    
+    mask_image = np.array(Image.open(image_path))
     
     token_texts = []
     token_scores = []