Update app.py

app.py

@@ -8,9 +8,10 @@ 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")
+from huggingface_hub import hf_hub_url, cached_download
 
+checkpoint_url = hf_hub_url(repo_id="osanseviero/magma", filename="model.pt")
+checkpoint_path = cached_download(config_file_url)
 
 model = Magma.from_checkpoint(
     config_path = "configs/MAGMA_v1.yml",