Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -167,8 +167,8 @@ def generate_wordcloud(entities: list, color_map: dict) -> plt.Figure:
|
|
| 167 |
|
| 168 |
@spaces.GPU
|
| 169 |
def all(text: str) -> Tuple[gr.HighlightedText, gr.HighlightedText, int, int, float, go.Figure, go.Figure, plt.Figure]:
|
| 170 |
-
ner_output_bin = process_ner(text,
|
| 171 |
-
ner_output_ext = process_ner(text,
|
| 172 |
|
| 173 |
binary_entities = [{"entity": ent["entity"], "score": ent["score"]} for ent in ner_output_bin["entities"]]
|
| 174 |
extended_entities = [{"entity": ent["entity"], "score": ent["score"]} for ent in ner_output_ext["entities"]]
|
|
|
|
| 167 |
|
| 168 |
@spaces.GPU
|
| 169 |
def all(text: str) -> Tuple[gr.HighlightedText, gr.HighlightedText, int, int, float, go.Figure, go.Figure, plt.Figure]:
|
| 170 |
+
ner_output_bin = process_ner(text, pipe_bin)
|
| 171 |
+
ner_output_ext = process_ner(text, pipe_ext)
|
| 172 |
|
| 173 |
binary_entities = [{"entity": ent["entity"], "score": ent["score"]} for ent in ner_output_bin["entities"]]
|
| 174 |
extended_entities = [{"entity": ent["entity"], "score": ent["score"]} for ent in ner_output_ext["entities"]]
|