Update app.py
Browse files
app.py
CHANGED
|
@@ -28,8 +28,7 @@ css = """
|
|
| 28 |
"""
|
| 29 |
|
| 30 |
# ====== 设备选择 ======
|
| 31 |
-
DEVICE = "cuda"
|
| 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()
|