Spaces:
Sleeping
Sleeping
Anigor66
commited on
Commit
·
d55bfe0
1
Parent(s):
6a97b56
fixed indentation
Browse files
app.py
CHANGED
|
@@ -73,7 +73,7 @@ print(f"Loading SAM model ({MODEL_TYPE}) from downloaded checkpoint...")
|
|
| 73 |
try:
|
| 74 |
# Ensure we always load onto CPU when no GPU is available
|
| 75 |
torch.load = patched_torch_load
|
| 76 |
-
sam = sam_model_registry[MODEL_TYPE](checkpoint=MODEL_CHECKPOINT)
|
| 77 |
finally:
|
| 78 |
torch.load = original_torch_load
|
| 79 |
|
|
|
|
| 73 |
try:
|
| 74 |
# Ensure we always load onto CPU when no GPU is available
|
| 75 |
torch.load = patched_torch_load
|
| 76 |
+
sam = sam_model_registry[MODEL_TYPE](checkpoint=MODEL_CHECKPOINT)
|
| 77 |
finally:
|
| 78 |
torch.load = original_torch_load
|
| 79 |
|