Hugging Face's logo

Spaces:
RaynWu2002
/
DORNet
Runtime error

App Files Community
1
RaynWu2002 commited on
Commit
2bf18ec
·
verified ·
1 Parent(s): ad51c09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ print("=" * 50)
33
 
34
  # init
35
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
36
- # device = "cpu"
37
  net = Net(tiny_model=False).to(device)
38
  model_ckpt_map = {
39
  "RGB-D-D": "./checkpoints/RGBDD.pth",
 
33
 
34
  # init
35
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
36
+ device = "cpu"
37
  net = Net(tiny_model=False).to(device)
38
  model_ckpt_map = {
39
  "RGB-D-D": "./checkpoints/RGBDD.pth",