Spaces:
Running on Zero
Running on Zero
Commit ·
3f835d3
1
Parent(s): c022cd6
run_segmentation on CPU
Browse files
app.py
CHANGED
|
@@ -894,7 +894,7 @@ if __name__ == '__main__':
|
|
| 894 |
dpt_pack = None
|
| 895 |
generated_object_map = {}
|
| 896 |
|
| 897 |
-
DEVICE = "
|
| 898 |
|
| 899 |
# Prepare models
|
| 900 |
## Grounding SAM
|
|
|
|
| 894 |
dpt_pack = None
|
| 895 |
generated_object_map = {}
|
| 896 |
|
| 897 |
+
DEVICE = "cpu"
|
| 898 |
|
| 899 |
# Prepare models
|
| 900 |
## Grounding SAM
|