cmarkea
/

dit-base-layout-detection

Image Segmentation

Inference Endpoints

Model card Files Files and versions Community

Cyrile commited on Aug 20

Commit

5853fa8

•

1 Parent(s): 8473981

Update README.md

Files changed (1) hide show

README.md +2 -0

README.md CHANGED Viewed

@@ -59,6 +59,8 @@ model = AutoModel.from_pretrained(
     "cmarkea/dit-base-layout-detection"
 )
 with torch.inference_mode():
     input_ids = img_proc(img, return_tensors='pt')
     output = model(**input_ids)

     "cmarkea/dit-base-layout-detection"
 )
+img: PIL.Image
 with torch.inference_mode():
     input_ids = img_proc(img, return_tensors='pt')
     output = model(**input_ids)