yabramuvdi commited on
Commit
db4c5ab
·
verified ·
1 Parent(s): bb934b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -70,8 +70,9 @@ def obtener_predicciones(texto, nombre_modelo, top_k=10):
70
  return top_k_tokens, top_k_prob.cpu().tolist()
71
 
72
  def generar_barplot(tokens, probabilidades):
73
- """Genera una gráfica de barras de Gradio con las palabras más probables."""
74
- datos = {"Palabra": tokens, "Probabilidad": probabilidades}
 
75
  return datos
76
 
77
  def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
@@ -134,10 +135,10 @@ with gr.Blocks() as demo:
134
 
135
  with gr.Row():
136
  barplot_resultados = gr.BarPlot(
137
- label="📊 Palabras más probables",
138
  x="Palabra",
139
  y="Probabilidad",
140
- title="Predicciones del modelo"
141
  )
142
 
143
  # Acciones de botones
 
70
  return top_k_tokens, top_k_prob.cpu().tolist()
71
 
72
  def generar_barplot(tokens, probabilidades):
73
+ """Genera una lista de diccionarios para Gradio BarPlot."""
74
+ datos = [{"Palabra": t, "Probabilidad": p} for t, p in zip(tokens, probabilidades)]
75
+ print(datos)
76
  return datos
77
 
78
  def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
 
135
 
136
  with gr.Row():
137
  barplot_resultados = gr.BarPlot(
138
+ value=[], # Se inicializa vacío
139
  x="Palabra",
140
  y="Probabilidad",
141
+ title="📊 Predicciones del modelo"
142
  )
143
 
144
  # Acciones de botones