remove cv2 import

app.py

@@ -1,7 +1,6 @@
 import os
 import time
 
-import cv2
 import gradio as gr
 import numpy as np
 import spaces
@@ -41,16 +40,6 @@ def annotate_image(input_image, detections, labels) -> np.ndarray:
     return output_image
 
 
-def resize_to_max_side(frame: np.ndarray, max_side: int = 640):
-    h, w = frame.shape[:2]
-    if h > w:
-        new_h, new_w = max_side, int(w * max_side / h)
-    else:
-        new_h, new_w = int(h * max_side / w), max_side
-
-    return cv2.resize(frame, (new_w, new_h))
-
-
 @spaces.GPU
 def process_video(
     input_video,
@@ -102,8 +91,10 @@ def process_video(
 
 
 def query(frame, classes, confidence_threshold, max_side=360):
-    frame_resized = resize_to_max_side(frame, max_side=max_side)
-    image = Image.fromarray(cv2.cvtColor(frame_resized, cv2.COLOR_BGR2RGB))
+    frame_resized = sv.resize_image(
+        image=frame, resolution_wh=(max_side, max_side), keep_aspect_ratio=True
+    )
+    image = Image.fromarray(frame_resized)
     inputs = processor(images=image, text=classes, return_tensors="pt").to(device)
     with torch.no_grad():
         start = time.time()

requirement.txt

@@ -1,6 +1,5 @@
 torch
 timm
-opencv-python
 git+https://github.com/yonigozlan/transformers.git@add-om-det-turbo
 supervision
 spaces