Gradio is a library for doing out-of-the-box visualizations for ML models: https://gradio.app/

It uses FastAPI under the hood, so should be straightforward to integrate w/ our FastAPI support.