srjosueaaron commited on
Commit
a805ecb
·
verified ·
1 Parent(s): a9b8668

Upload 3 files

Browse files
Files changed (3) hide show
  1. README.md +17 -13
  2. app.py +40 -0
  3. requirements.txt +4 -0
README.md CHANGED
@@ -1,13 +1,17 @@
1
- ---
2
- title: Fill Mask Demo
3
- emoji: 🏆
4
- colorFrom: gray
5
- colorTo: blue
6
- sdk: gradio
7
- sdk_version: 5.21.0
8
- app_file: app.py
9
- pinned: false
10
- license: mit
11
- ---
12
-
13
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
1
+ ---
2
+ title: "Modelo Fill-Mask en Español"
3
+ emoji: "🧠"
4
+ colorFrom: "purple"
5
+ colorTo: "blue"
6
+ sdk: "gradio"
7
+ short_description: "Completa oraciones con la palabra que falta."
8
+ models:
9
+ - "dccuchile/bert-base-spanish-wwm-cased"
10
+ tags:
11
+ - "fill-mask"
12
+ - "nlp"
13
+ - "spanish"
14
+ - "transformers"
15
+ - "gradio"
16
+ sdk_version: 5.21.0
17
+ ---
app.py ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+ import gradio as gr
3
+
4
+ fill_mask = pipeline("fill-mask", model="dccuchile/bert-base-spanish-wwm-cased")
5
+
6
+ def complete_mask(sentence):
7
+ if "[MASK]" not in sentence:
8
+ return "⚠️ Por favor, incluye '[MASK]' en la oración."
9
+
10
+ results = fill_mask(sentence)
11
+
12
+ output = "\n".join(
13
+ [f"🔹 **{res['token_str']}** (Probabilidad: {res['score']:.2%})" for res in results
14
+ if res['token_str'].strip() and res['token_str'] not in ["[UNK]", "."]]
15
+ )
16
+
17
+ return output if output else "⚠️ No se encontraron predicciones válidas."
18
+
19
+ with gr.Blocks() as demo:
20
+ gr.Markdown("""
21
+ # Modelo Fill-Mask en Español
22
+ Esta aplicación permite completar una oración con la palabra que falta, representada como '[MASK]'.
23
+ El modelo sugerirá las mejores opciones de palabras para completar la oración, con sus respectivas probabilidades.
24
+ """)
25
+
26
+ gr.Interface(
27
+ fn=complete_mask,
28
+ inputs=gr.Textbox(label="Oración con [MASK]", placeholder="Ejemplo: La capital de España es [MASK]."),
29
+ outputs=gr.Markdown(label="Predicciones"),
30
+ )
31
+
32
+ gr.Markdown("""
33
+ ---
34
+ Demostración de completado de palabras(máscaras) usando el modelo [dccuchile/bert-base-spanish-wwm-cased](https://huggingface.co/dccuchile/bert-base-spanish-wwm-cased).
35
+
36
+ Desarrollado con ❤️ por [@srjosueaaron](https://www.instagram.com/srjosueaaron/).
37
+ """)
38
+
39
+ if __name__ == "__main__":
40
+ demo.launch()
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ gradio
2
+ transformers[sentencepiece]
3
+ tensorflow
4
+ tf-keras