ZidongC commited on
Commit
5e61a51
·
verified ·
1 Parent(s): 251e8a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ css = """
28
  height: 62px;
29
  }
30
  """
31
- DEVICE = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu'
32
  config = 'panda_large.yaml'
33
  with open(config, 'r') as f:
34
  config = yaml.load(f, Loader=yaml.FullLoader)
 
28
  height: 62px;
29
  }
30
  """
31
+ DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
32
  config = 'panda_large.yaml'
33
  with open(config, 'r') as f:
34
  config = yaml.load(f, Loader=yaml.FullLoader)