Update app.py

app.py

@@ -98,7 +98,8 @@ 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")
+    args.device = "cuda" if torch.cuda.is_availble() else "cpu"
+    
     # Load the model
     model = AsymmetricMASt3R.from_pretrained(weights_path).to(args.device)
     chkpt_tag = hash_md5(weights_path)