cmarkea
/

detr-layout-detection

Image Segmentation

Inference Endpoints

Model card Files Files and versions Community

Cyrile commited on Aug 20

Commit

3fdd30a

•

1 Parent(s): 33deef6

Update README.md

Files changed (1) hide show

README.md +2 -0

README.md CHANGED Viewed

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

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