Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -85,6 +85,8 @@ elif torch.backends.mps.is_available():
|
|
| 85 |
device = torch.device("mps")
|
| 86 |
else:
|
| 87 |
device = torch.device("cpu")
|
|
|
|
|
|
|
| 88 |
print(f"using device: {device}")
|
| 89 |
|
| 90 |
# segmentation model
|
|
|
|
| 85 |
device = torch.device("mps")
|
| 86 |
else:
|
| 87 |
device = torch.device("cpu")
|
| 88 |
+
device = torch.device("cuda")
|
| 89 |
+
print(f"Force device to {device} due to ZeroGPU")
|
| 90 |
print(f"using device: {device}")
|
| 91 |
|
| 92 |
# segmentation model
|