Spaces:
Running
on
L4
Running
on
L4
Update app.py
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ def init_worker_sam(device='cuda'):
|
|
| 46 |
# Ideally, the SAM logic should be in a separate module to be picklable easily.
|
| 47 |
# But for this script, we can define the loading logic here.
|
| 48 |
|
| 49 |
-
checkpoint_path = 'models_v2/
|
| 50 |
|
| 51 |
# Load Model
|
| 52 |
try:
|
|
|
|
| 46 |
# Ideally, the SAM logic should be in a separate module to be picklable easily.
|
| 47 |
# But for this script, we can define the loading logic here.
|
| 48 |
|
| 49 |
+
checkpoint_path = 'models_v2/sam/sam_vit_b_01ec64.pth'
|
| 50 |
|
| 51 |
# Load Model
|
| 52 |
try:
|