Update app.py

app.py

@@ -71,9 +71,9 @@ def obtener_predicciones(texto, nombre_modelo, top_k=10):
     return top_k_tokens, top_k_prob.cpu().tolist()
 
 def generar_barplot(tokens, probabilidades):
-    """Convierte los datos en un DataFrame para Gradio BarPlot."""
+    """Convierte los datos en un DataFrame y lo ordena de mayor a menor probabilidad."""
     df = pd.DataFrame({"Palabra": tokens, "Probabilidad": probabilidades})
-    df.sort_values(by="Probabilidad", ascending=False, inplace=True)
+    df = df.sort_values(by="Probabilidad", ascending=False)  # ✅ Sort by probability (highest first)
     return df
 
 def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
@@ -83,7 +83,7 @@ def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
 
     tokens, probabilidades = obtener_predicciones(texto, nombre_modelo, int(top_k))
 
-    # Generar gráfico con Gradio BarPlot
+    # Generar gráfico con Gradio BarPlot (ahora ordenado)
     barplot_data = generar_barplot(tokens, probabilidades)
 
     return gr.update(choices=[f"'{t}'" for t in tokens]), barplot_data
@@ -97,18 +97,17 @@ def agregar_token_seleccionado(texto, token_seleccionado):
 
 # Crear la interfaz en español
 with gr.Blocks() as demo:
-    gr.Markdown("# 🔥 Predicción de Texto con Modelos de Lenguaje")
+    gr.Markdown("# 🔥 Predicción de Texto con Modelos Transformadores")
     gr.Markdown(
-        "Esta aplicación permite visualizar las probabilidades calculadas por un modelo de lenguaje en el proceso de generar texto."
+        "Esta aplicación permite generar palabras utilizando un modelo de lenguaje. "
         "Selecciona un modelo, introduce un texto y explora las palabras más probables a continuación."
-        "Puedes escoger una de las palabras con más probabilidad o escribir una palabra nueva y visualizar las nuevas predicciones del modelo."
     )
     
     with gr.Row():
         dropdown_modelo = gr.Dropdown(
             choices=list(AVAILABLE_MODELS.keys()),
             value="BLOOMZ-560M",
-            label="📌 Modelo de lenguaje (más opciones en el futuro)"
+            label="📌 Modelo de lenguaje"
         )
 
         dropdown_top_k = gr.Dropdown(
@@ -122,7 +121,7 @@ with gr.Blocks() as demo:
             lines=5,
             label="📝 Texto de entrada",
             placeholder="Escribe aquí...",
-            value="La mejor artista colombiana es"
+            value="Mi abuela me dejó una gran"
         )
     
     with gr.Row():