python中如何循环输入判断

如题， 从键盘中输入一个值，判断它是否位素数，如果是，打印出来，并且循环输入判断。

def prime(n):
    for i in range(2,n):
        if n%i==0:
            return False
    return True

while True:
    t = input()
    p = prime(int(t))
    if p: print('素数：',t)