Write a python program to classify the given input into two categories
Input: Apple
def classify(input):
    if input == "apple":
        label = "fruit"
    else:
        label = "not fruit"
    return label