from dual_attention.model_analysis import lm_visualization_app lm_visualization_app.run_app()