Update app.py

app.py

@@ -7,9 +7,13 @@ import re
 from magma import Magma
 from magma.image_input import ImageInput
 
+from huggingface_hub import hf_hub_download
+checkpoint_path = hf_hub_download(repo_id="osanseviero/magma", filename="model.pt")
+
+
 model = Magma.from_checkpoint(
     config_path = "configs/MAGMA_v1.yml",
-    checkpoint_path = "./mp_rank_00_model_states.pt",
+    checkpoint_path = checkpoint_path,
     device = 'cuda:0'
 )