Update app.py

app.py

@@ -70,8 +70,9 @@ def obtener_predicciones(texto, nombre_modelo, top_k=10):
     return top_k_tokens, top_k_prob.cpu().tolist()
 
 def generar_barplot(tokens, probabilidades):
-    """Genera una gráfica de barras de Gradio con las palabras más probables."""
-    datos = {"Palabra": tokens, "Probabilidad": probabilidades}
+    """Genera una lista de diccionarios para Gradio BarPlot."""
+    datos = [{"Palabra": t, "Probabilidad": p} for t, p in zip(tokens, probabilidades)]
+    print(datos)
     return datos
 
 def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
@@ -134,10 +135,10 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         barplot_resultados = gr.BarPlot(
-            label="📊 Palabras más probables",
+            value=[],  # Se inicializa vacío
             x="Palabra",
             y="Probabilidad",
-            title="Predicciones del modelo"
+            title="📊 Predicciones del modelo"
         )
 
     # Acciones de botones