zino36 commited on
Commit
4bfe017
1 Parent(s): aba1eee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ if __name__ == '__main__':
98
  server_name = args.server_name or ('0.0.0.0' if args.local_network else '127.0.0.1')
99
  weights_path = args.weights
100
 
101
- args.device = "cuda" if torch.cuda.is_availble() else "cpu"
102
 
103
  # Load the model
104
  model = AsymmetricMASt3R.from_pretrained(weights_path).to(args.device)
 
98
  server_name = args.server_name or ('0.0.0.0' if args.local_network else '127.0.0.1')
99
  weights_path = args.weights
100
 
101
+ args.device = "cuda" if torch.cuda.is_available() else "cpu"
102
 
103
  # Load the model
104
  model = AsymmetricMASt3R.from_pretrained(weights_path).to(args.device)