refactor(loading.py): add try-except block to handle KeyError when removing 'model' from params dictionary