update

README.md

@@ -1,5 +1,5 @@
 ---
-title: Borrowing Detection Es
+title: Borrowing Detection Español
 emoji: 🔍
 colorFrom: blue
 colorTo: purple

app.py

@@ -18,6 +18,7 @@ diplacy_dict_template = {
 def infer(input_text):
     displacy_ents = []
     borrowings = nlp(input_text)
+    
     for borrowing in borrowings:
         displacy_ent_dict = {
             "start": borrowing["start"],
@@ -28,9 +29,10 @@ def infer(input_text):
 
     colors = {"B-ENG": "linear-gradient(90deg, #aa9cfc, #fc9ce7)",
               "I-ENG": "linear-gradient(90deg, #99bfff, #a57cf0)",
-              "OTHER": "linear-gradient(90deg, #79d0a5, #f6e395)"}
+              "B-OTHER": "linear-gradient(90deg, #79d0a5, #f6e395)",
+              "I-OTHER": "linear-gradient(90deg, #f79a76, #fb6d6d)"}
     
-    options = {"ents": ["B-ENG", "I-ENG", "OTHER"], "colors": colors}
+    options = {"ents": ["B-ENG", "I-ENG", "B-OTHER", "I-OTHER"], "colors": colors}
     displacy_dict_template = {"text": input_text, "ents": displacy_ents, "title": None}
 
     html = displacy.render(displacy_dict_template, style="ent", page=True, manual=True, options=options)
@@ -44,12 +46,23 @@ def infer(input_text):
     return html
 
 
+description="""This space is a demo for the paper Detecting Unassimilated Borrowings in Spanish:
+[An Annotated Corpus and Approaches to Modeling](https://arxiv.org/pdf/2203.16169.pdf)
+
+The goal of the underlying model is to detect foreign words, e.g. anglicisms, in spanish texts.
+In general it has two types of tags for foreign words: *ENG* and *OTHER*. The authors used BIO-tagging,
+which is why in practice you will see a *B-* or *I-* in front of the tags.
+"""
+
 demo = gr.Interface(
     title="Borrowing Detection Español",
+    description=description,
     fn=infer,
     inputs=gr.Text(),
     outputs=gr.HTML(),
-    examples=["Buscamos data scientist para proyecto de machine learning."],
+    examples=["Buscamos data scientist para proyecto de machine learning.",
+              "Las fake news sobre la celebrity se reprodujeron por los 'mass media' en prime time.",
+              "Me gusta el cine noir y el anime."],
 )
 
 demo.launch()