factorial program

def factorial(n):
    if (n==0 or n==1):
        return 1
    else:
        return n * factorial(n-1)
n= int(input("enter the number"))
result = factorial(n)
print(result)