Update app.py

app.py

@@ -1,8 +1,6 @@
 from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline
 import gradio as gr
 
-
-
 def ner_tagging(text):
   model_name = "hackathon-pln-es/jurisbert-finetuning-ner"
   tokenizer = AutoTokenizer.from_pretrained(model_name,       add_prefix_space=True)
@@ -17,7 +15,7 @@ def ner_tagging(text):
   
   for i in range(len(text_2)):
       ent = ner_results[i]["entity"]
-      if ent != "LABEL_0":
+      if ent != "O":
           output.extend([(text_2[i], ent), (" ", None)])
       else:
           output.extend([(text_2[i], None), (" ", None)])