вы считываете данные из двух строк. А надо из одной.
Т.е. первые две строки надо заменить на одну строку "a, b = map(int, input().split())".