def factorial(n):
if n == 0 or n == 1:
return 1
else:
return n * factorial(n - 1)

Example usage:

num = 5
result = factorial(num)
print(f"The factorial of {num} is {result}")