Spaces:
Sleeping
Sleeping
Anigor66
commited on
Commit
·
17a1674
1
Parent(s):
94ac4ef
Fixed indentation
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ print(f"Loading SAM model ({MODEL_TYPE}) from downloaded checkpoint...")
|
|
| 58 |
|
| 59 |
try:
|
| 60 |
torch.load = patched_torch_load
|
| 61 |
-
|
| 62 |
finally:
|
| 63 |
torch.load = original_torch_load
|
| 64 |
|
|
@@ -484,12 +484,12 @@ def check_auto_mask_status():
|
|
| 484 |
"""
|
| 485 |
Check if automatic mask generation is available
|
| 486 |
"""
|
| 487 |
-
|
| 488 |
-
|
| 489 |
-
|
| 490 |
-
|
| 491 |
-
|
| 492 |
-
|
| 493 |
|
| 494 |
|
| 495 |
# =============================================================================
|
|
|
|
| 58 |
|
| 59 |
try:
|
| 60 |
torch.load = patched_torch_load
|
| 61 |
+
sam = sam_model_registry[MODEL_TYPE](checkpoint=MODEL_CHECKPOINT)
|
| 62 |
finally:
|
| 63 |
torch.load = original_torch_load
|
| 64 |
|
|
|
|
| 484 |
"""
|
| 485 |
Check if automatic mask generation is available
|
| 486 |
"""
|
| 487 |
+
return json.dumps({
|
| 488 |
+
'available': mask_generator is not None,
|
| 489 |
+
'model': MODEL_FILENAME if mask_generator else None,
|
| 490 |
+
'model_type': MODEL_TYPE,
|
| 491 |
+
'device': str(device)
|
| 492 |
+
})
|
| 493 |
|
| 494 |
|
| 495 |
# =============================================================================
|