Upload folder using huggingface_hub

README.md

@@ -1,12 +1,9 @@
 ---
-title: Spanish Bert Based NER
-emoji: 👀
-colorFrom: pink
-colorTo: pink
+title: Spanish-bert-based-NER
+emoji: 🔁
+colorFrom: blue
+colorTo: yellow
+app_file: app.py
 sdk: gradio
 sdk_version: 5.15.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---

app.py

@@ -0,0 +1,60 @@
+import gradio as gr
+from transformers import pipeline
+from collections import defaultdict
+
+ner_pipeline = pipeline("ner", model="syubraj/spanish_bert_based_ner")
+
+ENTITY_COLORS = {
+    "B-PER": "#FF5733",  
+    "I-PER": "#FF8D1A",  
+    "B-LOC": "#33FF57",  
+    "I-LOC": "#2EB82E",  
+    "B-ORG": "#3380FF",  
+    "I-ORG": "#1A66FF",  
+    "O": "#E0E0E0"       
+}
+
+def highlight_entities(text):
+    """
+    This function takes a Spanish text input and applies Named Entity Recognition (NER) using a pre-trained model.
+    The recognized entities are highlighted with distinct colors based on their categories.
+    A legend is also generated to indicate the meaning of each color.
+    
+    Parameters:
+    text (str): The input Spanish text.
+    
+    Returns:
+    str: The HTML formatted output with highlighted entities and a legend.
+    """
+    entities = ner_pipeline(text)
+    formatted_text = ""
+    last_idx = 0
+    
+    for ent in entities:
+        word = ent['word']
+        start, end = ent['start'], ent['end']
+        entity_label = ent['entity']
+        color = ENTITY_COLORS.get(entity_label, "#E0E0E0")
+        
+        formatted_text += text[last_idx:start] + f'<span style="background-color:{color}; padding:2px; border-radius:5px;">{word}</span>'
+        last_idx = end
+    
+    formatted_text += text[last_idx:]
+    
+    # Generate legend
+    legend_html = "<div><b>Legend:</b><br>"
+    for label, color in ENTITY_COLORS.items():
+        legend_html += f'<span style="background-color:{color}; padding:2px 5px; border-radius:5px; margin-right:5px;">{label}</span>'
+    legend_html += "</div><br>"
+    
+    return legend_html + formatted_text
+
+grn = gr.Interface(
+    fn=highlight_entities,
+    inputs=gr.Textbox(label="Enter Spanish Text"),
+    outputs=gr.HTML(label="NER Highlighted Text"),
+    title="Spanish Named Entity Recognition",
+    description="This interactive demo performs Named Entity Recognition (NER) on Spanish text. Recognized entities such as persons, locations, and organizations are highlighted in distinct colors for better readability. A legend is provided to help interpret the color coding.",
+)
+
+grn.launch()

requirements.txt

@@ -0,0 +1,2 @@
+gradio==5.15.0
+transformers==4.48.2