Add design file

app.py

@@ -28,9 +28,9 @@ pl.ion()
 torch.backends.cuda.matmul.allow_tf32 = True
 batch_size = 1
 inf = float('inf')
-weights_path = "checkpoints/geometry_pose.pth"
+# weights_path = "checkpoints/geometry_pose.pth"
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-ckpt = torch.load(weights_path, map_location=device)
+# ckpt = torch.load(weights_path, map_location=device)
 model = AsymmetricMASt3R(pos_embed='RoPE100', patch_embed_cls='ManyAR_PatchEmbed', img_size=(512, 512), head_type='catmlp+dpt', output_mode='pts3d+desc24', depth_mode=('exp', -inf, inf), conf_mode=('exp', 1, inf), enc_embed_dim=1024, enc_depth=24, enc_num_heads=16, dec_embed_dim=768, dec_depth=12, dec_num_heads=12, two_confs=True, desc_conf_mode=('exp', 0, inf))
 model = AsymmetricMASt3R.from_pretrained("zhang3z/FLARE").to(device)
 # model.from_pretrained(ckpt['model'], strict=False)