add model def import

app.py

@@ -1,3 +1,11 @@
+#### pull files from hub
+from huggingface_hub import hf_hub_download
+import os
+yaml_file_path=hf_hub_download(repo_id="ibm-nasa-geospatial/Prithvi-100M", filename="Prithvi_100M_config.yaml")
+checkpoint=hf_hub_download(repo_id="ibm-nasa-geospatial/Prithvi-100M", filename='Prithvi_100M.pt')
+model_def=hf_hub_download(repo_id="ibm-nasa-geospatial/Prithvi-100M", filename='Prithvi.py')
+os.system(f'cp {model_def} .')
+#####
 import argparse
 import functools
 import os
@@ -363,10 +371,8 @@ def predict_on_images(data_files: list, mask_ratio: float, yaml_file_path: str,
 
     return outputs
 
-from huggingface_hub import hf_hub_download
 
-yaml_file_path=hf_hub_download(repo_id="ibm-nasa-geospatial/Prithvi-100M", filename="Prithvi_100M_config.yaml")
-checkpoint=hf_hub_download(repo_id="ibm-nasa-geospatial/Prithvi-100M", filename='Prithvi_100M.pt')
+
 
 func = partial(predict_on_images, yaml_file_path=yaml_file_path,checkpoint=checkpoint)