Update README.md
Browse files
README.md
CHANGED
|
@@ -19,7 +19,7 @@ from transformers import pipeline
|
|
| 19 |
|
| 20 |
segment = Volume(20230827161847, normalize=True)
|
| 21 |
|
| 22 |
-
classifier = pipeline(model="
|
| 23 |
prediction = classifier(np.transpose(segment[17:43, 200:5600,1000:4600],(1,2,0)).astype(np.float32)) # prediction shape ~ (512,512)
|
| 24 |
|
| 25 |
plt.imshow(prediction)
|
|
|
|
| 19 |
|
| 20 |
segment = Volume(20230827161847, normalize=True)
|
| 21 |
|
| 22 |
+
classifier = pipeline(model="scrollprize/ink_detection_pipeline", trust_remote_code=True, tile_size=64, stride=32, scale_factor=16)
|
| 23 |
prediction = classifier(np.transpose(segment[17:43, 200:5600,1000:4600],(1,2,0)).astype(np.float32)) # prediction shape ~ (512,512)
|
| 24 |
|
| 25 |
plt.imshow(prediction)
|