runtime error

Exit code: 1. Reason: te-packages/gradio/blocks.py", line 1520, in call_function prediction = await anyio.to_thread.run_sync( # type: ignore File "/usr/local/lib/python3.10/site-packages/anyio/to_thread.py", line 56, in run_sync return await get_async_backend().run_sync_in_worker_thread( File "/usr/local/lib/python3.10/site-packages/anyio/_backends/_asyncio.py", line 2505, in run_sync_in_worker_thread return await future File "/usr/local/lib/python3.10/site-packages/anyio/_backends/_asyncio.py", line 1005, in run result = context.run(func, *args) File "/usr/local/lib/python3.10/site-packages/gradio/utils.py", line 826, in wrapper response = f(*args, **kwargs) File "/home/user/app/app.py", line 100, in fn_url_with_download processed_image, origin, png_image = fn(im) File "/home/user/app/app.py", line 66, in fn processed_image = process(im) File "/home/user/app/app.py", line 53, in process input_images = transform_image(image).unsqueeze(0).to("cpu") # GPU -> CPU로 변경 File "/usr/local/lib/python3.10/site-packages/torchvision/transforms/transforms.py", line 95, in __call__ img = t(img) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/torchvision/transforms/transforms.py", line 277, in forward return F.normalize(tensor, self.mean, self.std, self.inplace) File "/usr/local/lib/python3.10/site-packages/torchvision/transforms/functional.py", line 350, in normalize return F_t.normalize(tensor, mean=mean, std=std, inplace=inplace) File "/usr/local/lib/python3.10/site-packages/torchvision/transforms/_functional_tensor.py", line 928, in normalize return tensor.sub_(mean).div_(std) RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0

Container logs: