Anigor66 commited on
Commit
17a1674
·
1 Parent(s): 94ac4ef

Fixed indentation

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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
- sam = sam_model_registry[MODEL_TYPE](checkpoint=MODEL_CHECKPOINT)
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
- 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
  # =============================================================================
 
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
  # =============================================================================