Update app.py

app.py

@@ -58,7 +58,7 @@ def obtener_predicciones(texto, nombre_modelo, top_k=10):
     if current_model_name != nombre_modelo:
         cargar_modelo(nombre_modelo)
     
-    entradas = current_tokenizer(texto, return_tensors="pt").to(device)
+    entradas = current_tokenizer(texto.strip(), return_tensors="pt").to(device)
 
     with torch.no_grad():
         salidas = current_model(**entradas)
@@ -66,20 +66,19 @@ def obtener_predicciones(texto, nombre_modelo, top_k=10):
         probabilidades = torch.nn.functional.softmax(logits, dim=-1)
     
     top_k_prob, top_k_indices = torch.topk(probabilidades, k=top_k)
-    top_k_tokens = [current_tokenizer.decode([idx.item()]) for idx in top_k_indices]
+    top_k_tokens = [current_tokenizer.decode([idx.item()]).strip() for idx in top_k_indices]  # ✅ Strip spaces
     
     return top_k_tokens, top_k_prob.cpu().tolist()
 
 def generar_barplot(tokens, probabilidades):
     """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
+    return df
 
 def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
     """Obtiene predicciones y actualiza la UI."""
     if token_custom:
-        texto += token_custom
+        texto = texto.rstrip() + " " + token_custom.strip()  # ✅ Prevents extra whitespaces
 
     tokens, probabilidades = obtener_predicciones(texto, nombre_modelo, int(top_k))
 
@@ -89,10 +88,10 @@ def predecir_siguiente_palabra(nombre_modelo, texto, top_k, token_custom=""):
     return gr.update(choices=[f"'{t}'" for t in tokens]), barplot_data
 
 def agregar_token_seleccionado(texto, token_seleccionado):
-    """Agrega el token seleccionado al texto de entrada."""
+    """Agrega el token seleccionado al texto de entrada sin espacios extra."""
     if token_seleccionado:
-        token_limpio = token_seleccionado.strip("'")
-        texto += f" {token_limpio}"
+        token_limpio = token_seleccionado.strip("'").strip()  # ✅ Removes unwanted spaces
+        texto = texto.rstrip() + " " + token_limpio  # ✅ Ensures no double spaces
     return texto
 
 # Crear la interfaz en español
@@ -137,9 +136,10 @@ with gr.Blocks() as demo:
     with gr.Row():
         barplot_resultados = gr.BarPlot(
             value=pd.DataFrame(columns=["Palabra", "Probabilidad"]),  # ✅ Empty DataFrame to initialize
-            x="Palabra",
-            y="Probabilidad",
-            title="📊 Predicciones del modelo"
+            x="Probabilidad",  # ✅ Swap axes to make it horizontal
+            y="Palabra",
+            title="📊 Predicciones del modelo",
+            orientation="h"  # ✅ Makes the barplot horizontal
         )
 
     # Acciones de botones