moyanxinxu commited on
Commit
6f7501d
1 Parent(s): d3b1997

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ import supervision as sv
3
  from func import detect_and_track
4
  from transformers import AutoImageProcessor, AutoModelForObjectDetection
5
 
6
- processor = AutoImageProcessor.from_pretrained("PekingU/rtdetr_r50vd_coco_o365")
7
- model = AutoModelForObjectDetection.from_pretrained("PekingU/rtdetr_r50vd_coco_o365")
8
  tracker = sv.ByteTrack()
9
 
10
  mask_annotator = sv.MaskAnnotator()
 
3
  from func import detect_and_track
4
  from transformers import AutoImageProcessor, AutoModelForObjectDetection
5
 
6
+ processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
7
+ model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
8
  tracker = sv.ByteTrack()
9
 
10
  mask_annotator = sv.MaskAnnotator()