Spaces:

projecte-aina
/

multiner_demo

Runtime error

crodri commited on Oct 24, 2023

Commit

1652c9d

1 Parent(s): ecf1622

changed aggregation strategy

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from AinaTheme import AinaGradioTheme
 from transformers import pipeline
 import gradio as gr
-ner_pipeline = pipeline("token-classification", model="projecte-aina/multiner_ceil",aggregation_strategy="first")
 def ner(text):
     output = ner_pipeline(text)

 from transformers import pipeline
 import gradio as gr
+ner_pipeline = pipeline("token-classification", model="projecte-aina/multiner_ceil",aggregation_strategy="simple")
 def ner(text):
     output = ner_pipeline(text)