mshukor commited on
Commit
aff9f36
1 Parent(s): fb7cc0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ yaml=YAML(typ='safe')
43
 
44
 
45
  use_cuda = torch.cuda.is_available()
46
- device = torch.deivce('cuda') if use_cuda else torch.deivce('cpu')
47
 
48
  ## Load model
49
 
 
43
 
44
 
45
  use_cuda = torch.cuda.is_available()
46
+ device = torch.device('cuda') if use_cuda else torch.device('cpu')
47
 
48
  ## Load model
49