Viết chương trình nhập vào số tự nhiên N rồi thông báo lên màn hình số đó có phải là số nguyên tố hay không. Lưu ý: Python

def check(n):
    if n<2:
        return False
    for i in range(2,int(n**0.5+1)):
        if n%i==0:
            return False
    return True

n = int(input())
print(f'{n} la so nguyen to' if check(n) else f'{n} khong phai la so nguyen to')

Của em đây nhé.
def is_prime(n):
    if n <= 1:
        return False
    for i in range(2, int(n**0.5) + 1):
        if n % i == 0:
            return False
    return True
N = int(input("Nhập vào một số tự nhiên N: "))
if is_prime(N):
    print(f"{N} là số nguyên tố.")
else:
    print(f"{N} không là số nguyên tố.")