Update app.py

app.py

@@ -70,10 +70,10 @@ 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 lista de diccionarios para Gradio BarPlot."""
-    datos = [{"Palabra": t, "Probabilidad": p} for t, p in zip(tokens, probabilidades)]
-    print(datos)
-    return datos
+    """Convierte los datos en un DataFrame para Gradio BarPlot."""
+    df = pd.DataFrame({"Palabra": tokens, "Probabilidad": probabilidades})
+    print(df)
+    return df  # ✅ Now returning a Pandas DataFrame instead of a list
 
 def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
     """Obtiene predicciones y actualiza la UI."""
@@ -135,7 +135,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         barplot_resultados = gr.BarPlot(
-            value=[],  # Se inicializa vacío
+            value=pd.DataFrame(columns=["Palabra", "Probabilidad"]),  # ✅ Empty DataFrame to initialize
             x="Palabra",
             y="Probabilidad",
             title="📊 Predicciones del modelo"