stevenbucaille commited on
Commit
41ce12f
·
1 Parent(s): 8680cad

card: replace official repo name from stevenbucaille to AnnaZhang

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -33,7 +33,7 @@ Training Details:
33
 
34
  ### How to use
35
 
36
- You can use the raw model for object detection. See the [model hub](https://huggingface.co/models?search=stevenbucaille/lw-detr) to look for all available LW DETR models.
37
 
38
  Here is how to use this model:
39
 
@@ -46,8 +46,8 @@ import requests
46
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
47
  image = Image.open(requests.get(url, stream=True).raw)
48
 
49
- processor = AutoImageProcessor.from_pretrained("stevenbucaille/lwdetr_tiny_30e_objects365")
50
- model = LwDetrForObjectDetection.from_pretrained("stevenbucaille/lwdetr_tiny_30e_objects365")
51
 
52
  inputs = processor(images=image, return_tensors="pt")
53
  outputs = model(**inputs)
 
33
 
34
  ### How to use
35
 
36
+ You can use the raw model for object detection. See the [model hub](https://huggingface.co/models?search=AnnaZhang/lw-detr) to look for all available LW DETR models.
37
 
38
  Here is how to use this model:
39
 
 
46
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
47
  image = Image.open(requests.get(url, stream=True).raw)
48
 
49
+ processor = AutoImageProcessor.from_pretrained("AnnaZhang/lwdetr_tiny_30e_objects365")
50
+ model = LwDetrForObjectDetection.from_pretrained("AnnaZhang/lwdetr_tiny_30e_objects365")
51
 
52
  inputs = processor(images=image, return_tensors="pt")
53
  outputs = model(**inputs)