Dingning commited on
Commit
8e51806
·
verified ·
1 Parent(s): f5a7864

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -28,8 +28,7 @@ css = """
28
  """
29
 
30
  # ====== 设备选择 ======
31
- DEVICE = "cuda" if torch.cuda.is_available() else \
32
- "mps" if torch.backends.mps.is_available() else "cpu"
33
 
34
  # ====== 模型加载 ======
35
  model = Bridge()
 
28
  """
29
 
30
  # ====== 设备选择 ======
31
+ DEVICE = "cuda" else "cpu"
 
32
 
33
  # ====== 模型加载 ======
34
  model = Bridge()