Help me in solving DIFFSUM problem

My issue

In the input format it says n1 and n2 will be readed two different lines.
But, sample example boyh are readed in single line
I tried both but it shows error for both the cases

My code

a = int(input())
b = int(input())
if a > b:
    print( a - b )
else:
    print( a + b )

Problem Link: DIFFSUM Problem - CodeChef