Vidit2003 commited on
Commit
1e61d09
·
verified ·
1 Parent(s): 36b957a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -92,7 +92,7 @@ class PerImageNormalize(nn.Module):
92
 
93
  # Load model
94
  device = "cuda"
95
- model = AutoModel.from_pretrained("CaicedoLab/CHAMMI-75", trust_remote_code=True)
96
  model.to(device).eval()
97
 
98
  # Define transforms
 
92
 
93
  # Load model
94
  device = "cuda"
95
+ model = AutoModel.from_pretrained("CaicedoLab/MorphEm", trust_remote_code=True)
96
  model.to(device).eval()
97
 
98
  # Define transforms