Update app.py

app.py

@@ -74,7 +74,7 @@ def process_ner(text: str, pipeline) -> dict:
 
     return {"text": text, "entities": entities}
 
-def generate_charts(ner_output_bin: dict, ner_output_ext: dict) -> Tuple[go.Figure, go.Figure, np.ndarray]:
+def generate_charts(ner_output_ext: dict) -> Tuple[go.Figure, np.ndarray]:
     entities_ext = [entity['entity'] for entity in ner_output_ext['entities']]
     
     # Counting entities for extended classification
@@ -149,7 +149,7 @@ def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str]) -> np.nd
 def all(text: str):
     ner_output_ext = process_ner(text, pipe_ext)
     
-    pie_chart, wordcloud_image = generate_charts(ner_output_bin, ner_output_ext)
+    pie_chart, wordcloud_image = generate_charts(ner_output_ext)
     
     return (ner_output_ext, pie_chart, wordcloud_image)