test

app.py

@@ -4,18 +4,19 @@ import cv2
 import tempfile
 from ultralytics import YOLOv10
 
-model = YOLOv10.from_pretrained(f'jameslahm/yolov10n')
+image_processor = RTDetrImageProcessor.from_pretrained("PekingU/rtdetr_r50vd")
+model = RTDetrForObjectDetection.from_pretrained("PekingU/rtdetr_r50vd")
 
 @spaces.GPU
 def yolov10_inference(image, conf_threshold):
-    width, _ = image.size
-    import time
-    start = time.time()
-    results = model.predict(source=image, imgsz=width, conf=conf_threshold)
-    end = time.time()
-    print("time", end - start)
-    annotated_image = results[0].plot()
-    return annotated_image[:, :, ::-1]
+
+    inputs = image_processor(images=image, return_tensors="pt")
+
+    with torch.no_grad():
+        outputs = model(**inputs)
+
+    results = image_processor.post_process_object_detection(outputs, target_sizes=torch.tensor([image.size[::-1]]), threshold=0.3)
+
 
 
 def app():