Hugging Face's logo

Spaces:
ArtificialCoder02
/
pmf_zeroshot
Runtime error

App Files Community
ArtificialCoder02 commited on
Commit
59d9031
·
1 Parent(s): 3b7c783

Made Changes

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ args.cx = '772f0ccc480364116'
25
 
26
 
27
  # model
28
- device = 'cpu' #torch.device("cuda" if torch.cuda.is_available() else "cpu")
29
  model = get_model(args)
30
  model.to(device)
31
  checkpoint = torch.hub.load_state_dict_from_url(args.resume, map_location='cuda')
 
25
 
26
 
27
  # model
28
+ device = 'cuda' #torch.device("cuda" if torch.cuda.is_available() else "cpu")
29
  model = get_model(args)
30
  model.to(device)
31
  checkpoint = torch.hub.load_state_dict_from_url(args.resume, map_location='cuda')