"Ошибка: выход алгоритма превысил максимально допустимый размер вывода 16883260 > 15728640 байт"

Question

Я пытаюсь развернуть модель Pytorch на алгоритме (платформа MLOps) и вернуть строку json в качестве вывода. Я определил функцию, которая предсказывает маску объекта по входному изображению с использованием модели и возвращает массив оценок предсказания. Я преобразовал его в json, используя data = { 'pred': json.dumps(pred.tolist()) }а потом . Когда я хочу вернуться dataили же json_stringон показывает вышеуказанную ошибку.

Я попытался протестировать это на моем локальном компьютере. Когда я печатаю json_string = json.dumps(data)он не выдает ошибку, хотя печатает этот огромный вывод консоли в течение нескольких секунд и завершает выполнение. Я не уверен, как изменить этот код для развертывания.