yabramuvdi commited on
Commit
2ba7738
·
verified ·
1 Parent(s): 22ac25d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -73,6 +73,7 @@ def obtener_predicciones(texto, nombre_modelo, top_k=10):
73
  def generar_barplot(tokens, probabilidades):
74
  """Convierte los datos en un DataFrame para Gradio BarPlot."""
75
  df = pd.DataFrame({"Palabra": tokens, "Probabilidad": probabilidades})
 
76
  return df
77
 
78
  def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
@@ -96,17 +97,18 @@ def agregar_token_seleccionado(texto, token_seleccionado):
96
 
97
  # Crear la interfaz en español
98
  with gr.Blocks() as demo:
99
- gr.Markdown("# 🔥 Predicción de Texto con Modelos Transformadores")
100
  gr.Markdown(
101
- "Esta aplicación permite generar palabras utilizando un modelo de lenguaje. "
102
  "Selecciona un modelo, introduce un texto y explora las palabras más probables a continuación."
 
103
  )
104
 
105
  with gr.Row():
106
  dropdown_modelo = gr.Dropdown(
107
  choices=list(AVAILABLE_MODELS.keys()),
108
  value="BLOOMZ-560M",
109
- label="📌 Modelo de lenguaje"
110
  )
111
 
112
  dropdown_top_k = gr.Dropdown(
@@ -120,7 +122,7 @@ with gr.Blocks() as demo:
120
  lines=5,
121
  label="📝 Texto de entrada",
122
  placeholder="Escribe aquí...",
123
- value="Mi abuela me dejó una gran"
124
  )
125
 
126
  with gr.Row():
 
73
  def generar_barplot(tokens, probabilidades):
74
  """Convierte los datos en un DataFrame para Gradio BarPlot."""
75
  df = pd.DataFrame({"Palabra": tokens, "Probabilidad": probabilidades})
76
+ df.sort_values(by="Probabilidad", ascending=False, inplace=True)
77
  return df
78
 
79
  def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
 
97
 
98
  # Crear la interfaz en español
99
  with gr.Blocks() as demo:
100
+ gr.Markdown("# 🔥 Predicción de Texto con Modelos de Lenguaje")
101
  gr.Markdown(
102
+ "Esta aplicación permite visualizar las probabilidades calculadas por un modelo de lenguaje en el proceso de generar texto."
103
  "Selecciona un modelo, introduce un texto y explora las palabras más probables a continuación."
104
+ "Puedes escoger una de las palabras con más probabilidad o escribir una palabra nueva y visualizar las nuevas predicciones del modelo."
105
  )
106
 
107
  with gr.Row():
108
  dropdown_modelo = gr.Dropdown(
109
  choices=list(AVAILABLE_MODELS.keys()),
110
  value="BLOOMZ-560M",
111
+ label="📌 Modelo de lenguaje (más opciones en el futuro)"
112
  )
113
 
114
  dropdown_top_k = gr.Dropdown(
 
122
  lines=5,
123
  label="📝 Texto de entrada",
124
  placeholder="Escribe aquí...",
125
+ value="La mejor artista colombiana es"
126
  )
127
 
128
  with gr.Row():