Fix import error

app.py

@@ -7,8 +7,6 @@ import gradio as gr
 from einops import rearrange
 from functools import partial
 from huggingface_hub import hf_hub_download
-from prithvi_mae import PrithviMAE
-from inference import process_channel_group, _convert_np_uint8, load_example, run_model
 
 # pull files from hub
 token = os.environ.get("HF_TOKEN", None)
@@ -23,6 +21,8 @@ model_inference = hf_hub_download(repo_id="ibm-nasa-geospatial/Prithvi-EO-2.0-30
 os.system(f'cp {model_def} .')
 os.system(f'cp {model_inference} .')
 
+from prithvi_mae import PrithviMAE
+from inference import process_channel_group, _convert_np_uint8, load_example, run_model
 
 def extract_rgb_imgs(input_img, rec_img, mask_img, channels, mean, std):
     """ Wrapper function to save Geotiff images (original, reconstructed, masked) per timestamp.