Update README.md
Browse files
README.md
CHANGED
|
@@ -92,7 +92,7 @@ class PerImageNormalize(nn.Module):
|
|
| 92 |
|
| 93 |
# Load model
|
| 94 |
device = "cuda"
|
| 95 |
-
model = AutoModel.from_pretrained("CaicedoLab/
|
| 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
|