Spaces:
Sleeping
Sleeping
dennistrujillo
commited on
Update app.py
Browse files
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']
|
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
|