It seems the default in_channels value is set to 2, while your input data has only a single channel.
Set in_channels=1 in your model initialization and run the code again.
PS: It’s always better to post directly by wrapping it in three backticks ```