Change import of MAE VIT (#10)

- Change import of MAE VIT (7dc0c59fc33d36274ac225e2b8c023594c07193e)


Co-authored-by: Elias Hartig <[email protected]>

Prithvi_run_inference.py

@@ -9,8 +9,7 @@ import torch
 import yaml
 from einops import rearrange
 
-from mae.models_mae import MaskedAutoencoderViT
-
+from Prithvi import MaskedAutoencoderViT
 
 NO_DATA = -9999
 NO_DATA_FLOAT = 0.0001