fastapi pydantic keras h5py torch albumentations transformers pytorch-lightning torchmetrics torchvision numpy uvicorn pydantic