Update app.py

app.py

@@ -98,6 +98,7 @@ if __name__ == '__main__':
     server_name = args.server_name or ('0.0.0.0' if args.local_network else '127.0.0.1')
     weights_path = args.weights
 
+    args.device = torch.device("cuda" if use_cuda else "cpu")
     # Load the model
     model = AsymmetricMASt3R.from_pretrained(weights_path).to(args.device)
     chkpt_tag = hash_md5(weights_path)