Functions to save machine learning Models as .pickle files and importing them

pickle.py

import pickle
def save_pickle(model):
    name=input('Name of the model: ')
    with open(f'{name}.pickle','wb') as model:
        pickle.dump(rf_model,model)
    print('Model Saved.')

def load_pickle(filename):
    file = open(filename,'rb')
    model = pickle.load(file)
    return model
  
'''
Usage:
  To save a model to the pickle file:
     model = model.fit(x,y....)
     save_to_pickle(model)
   To use a pickled model
      model = load_pickle('path/to/pickle')
'''