legend fix

interfaces/ner.py

@@ -34,10 +34,10 @@ def named_entity_recognition(text, language):
  pipeline = huspacy.load() if model_id.startswith("hu") else spacy.load(model_id)
  doc = pipeline(text)
  entities = [{"entity":ent.label_, "start":ent.start_char, "end":ent.end_char} for ent in doc.ents]
- legend = [f'<li> <b>{ent}</b> = <i>{spacy.glossary.GLOSSARY[ent]}</i> </li>' for ent in set(doc.ents) ]
+ legend = "<ul>"+"".join([f"<li> <b>{ent}</b> = <i>{spacy.glossary.GLOSSARY[ent]}</i> </li>" for ent in set(doc.ents)])+"</ul>"
  output = {"text":text, "entities":entities}
  model_id_hf = f"huspacy/{model_id}" if model_id.startswith("hu") else f"spacy/{model_id}"
- output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id_hf}">{model_id_hf}</a> model.</p> <ul> {" ".join(legend)} </ul>'
+ output_info = legend + f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id_hf}">{model_id_hf}</a> model.</p> <ul> {" ".join(legend)} </ul>'
  return output, output_info
 
 demo = gr.Interface(