Switching to huggingface model

demo.py

@@ -12,6 +12,7 @@ import tempfile
 
 import gradio
 import torch
+from huggingface_hub import hf_hub_download
 
 sys.path.append('src/mast3r_src')
 sys.path.append('src/mast3r_src/dust3r')
@@ -45,8 +46,6 @@ def get_reconstructed_scene(outdir, model, device, silent, image_size, ios_mode,
 
 if __name__ == '__main__':
 
-    weights_path = sys.argv[1]
-
     image_size = 512
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     server_name = '127.0.0.1'
@@ -55,6 +54,9 @@ if __name__ == '__main__':
     silent = False
     ios_mode = True
 
+    model_name = "brandonsmart/splatt3r_v1.0"
+    filename = "epoch=19-step=1200.ckpt"
+    weights_path = hf_hub_download(repo_id=model_name, filename=filename)
     model = main.MAST3RGaussians.load_from_checkpoint(weights_path, device)
     chkpt_tag = hash_md5(weights_path)