convert miles to km and vice versa

print("Convert: \n1. miles to km \n2. km to miles")

unit1 = "1. miles to km"
unit2 = "2. km to miles"
dis = float(input("Enter distance:    "))
unit = int(input("chose 1 or 2:   "))

if unit == 1:
    result = dis *  1.61
    print(result,"  km")
elif unit == 2:
    result = dis * 0.62
    print(result, "  mi")
else:
    print("invalid input")