Update app.py

app.py

@@ -25,11 +25,16 @@ def NextTokenPredictionApp():
         preds = predict_next_token(new_text)
         predictions.set(preds)
 
-    sol.InputText(value=text.value, on_change=on_text_change, placeholder="Escribe algo en español...")
+    sol.Markdown("# Predicción del Próximo Token")
+    sol.Markdown("Ingrese un texto en español y vea las predicciones para el próximo token.")
+    
+    sol.InputText(value=text.value, on_change=on_text_change, placeholder="Escribe algo en español...", fullwidth=True)
+    sol.Button("Predecir", on_click=lambda: on_text_change(text.value))
 
     if predictions.value:
-        sol.Markdown("### Predicciones de tokens:")
+        sol.Markdown("## Predicciones de tokens:")
         for token, prob in predictions.value:
-            sol.Markdown(f"- {token}: {prob:.4f}")
+            sol.Markdown(f"- **{token}**: {prob:.4f}")
 
-sol.run(NextTokenPredictionApp)
+if __name__ == "__main__":
+    sol.run(NextTokenPredictionApp, title="Next Token Prediction App")