Aryn
/

Object Detection
Transformers
Safetensors
deformable_detr
vision
Inference Endpoints
jonfritz commited on
Commit
3b46ecb
1 Parent(s): 2441def

Update README.md

Browse files

Changed model name

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -50,8 +50,8 @@ import requests
50
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
51
  image = Image.open(requests.get(url, stream=True).raw)
52
 
53
- processor = AutoImageProcessor.from_pretrained("Aryn/deformable-detr")
54
- model = DeformableDetrForObjectDetection.from_pretrained("Aryn/deformable-detr")
55
 
56
  inputs = processor(images=image, return_tensors="pt")
57
  outputs = model(**inputs)
 
50
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
51
  image = Image.open(requests.get(url, stream=True).raw)
52
 
53
+ processor = AutoImageProcessor.from_pretrained("Aryn/deformable-detr-DocLayNet")
54
+ model = DeformableDetrForObjectDetection.from_pretrained("Aryn/deformable-detr-DocLayNet")
55
 
56
  inputs = processor(images=image, return_tensors="pt")
57
  outputs = model(**inputs)