made the class names in model_loading.py compliant ...

... with the ones used in models.py