Update app.py

app.py

@@ -17,7 +17,7 @@ import spaces
 import supervision as sv
 import torch
 import tqdm
-from transformers import AutoProcessor, RTDetrForObjectDetection, VitPoseForPoseEstimation
+from transformers import AutoProcessor, AutoModelForObjectDetection
 
 DESCRIPTION = "# DAB-DETR"
 
@@ -27,7 +27,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 checkpoint = "IDEA-Research/dab-detr-resnet-50-dc5-pat3"
 image_processor = AutoProcessor.from_pretrained(checkpoint)
-model = RTDetrForObjectDetection.from_pretrained(checkpoint, device_map=device)
+model = AutoModelForObjectDetection.from_pretrained(checkpoint, device_map=device)
 
 
 @spaces.GPU(duration=5)