dennistrujillo commited on
Commit
de86f89
·
verified ·
1 Parent(s): 42aa5e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ def process_images(file, x_min, y_min, x_max, y_max):
75
  checkpoint = torch.load(model_checkpoint_path)
76
 
77
  # Create model instance
78
- medsam_model = sam_model_registry['vit_b']() # Create model instance without checkpoint
79
 
80
  # Load the state dictionary into the model
81
  medsam_model.load_state_dict(checkpoint) # Load the saved weights
 
75
  checkpoint = torch.load(model_checkpoint_path)
76
 
77
  # Create model instance
78
+ medsam_model = sam_model_registry['vit_b'] # Create model instance without checkpoint
79
 
80
  # Load the state dictionary into the model
81
  medsam_model.load_state_dict(checkpoint) # Load the saved weights