How to exit if statement

At the end of the code im trying to make inputs "2" or "3" restart the program and anything else end the program. However as im testing it seems any input for "next student ID" restarts it.

pass_mark = 40

print("Student IDs are: ")
for i in range(1,4):
    print(i)

def main():        

    selection = input("Please select a student ID: ")
    days_late = int(input("Please enter the number of days late: "))
    
    if 0<=days_late<=5:
        raw_mark = int(input("Please enter the raw mark: "))
        print("Raw Mark is: ",raw_mark,"/100")
        adjusted_grade = (raw_mark-days_late*5)
        
        if adjusted_grade<pass_mark:
            print("Grade has been capped at 40")
            print("\n")
            print("ID    Raw Mark    Days Late    Final Mark    Capping")
            print(selection,"   ",raw_mark,"        ",days_late,"          ","40","          ","Yes")
        
        else:        
            print("Grade after late submission adjustment is ",(adjusted_grade),"/100")
            print("\n")
            print("ID    Raw Mark    Days Late    Final Mark    Capping")
            print(selection,"   ",raw_mark,"        ",days_late,"          ",(adjusted_grade),"          ","no")
        

    elif days_late>5:
        print("As submission is over 5 days late, assignment is failed")

    else:
        print("Invalid Input")

    repeat = input("please enter the next student ID: ")
    if repeat == "2" or "3":
        main()
    else:
        exit()

main()