remove set image size

app.py

@@ -44,7 +44,6 @@ def process_video(
     input_video,
     confidence_threshold,
     classes,
-    max_side,
     progress=gr.Progress(track_tqdm=True),
 ):
     classes = classes.strip(" ").split(",")
@@ -61,9 +60,7 @@ def process_video(
                 frame = next(frame_generator)
             except StopIteration:
                 break
-            results, fps = query(
-                frame, classes, confidence_threshold, max_side=max_side
-            )
+            results, fps = query(frame, classes, confidence_threshold)
             all_fps.append(fps)
             detections = []
 
@@ -92,11 +89,8 @@ def process_video(
     )
 
 
-def query(frame, classes, confidence_threshold, max_side=360):
-    frame_resized = sv.resize_image(
-        image=frame, resolution_wh=(max_side, max_side), keep_aspect_ratio=True
-    )
-    image = Image.fromarray(frame_resized)
+def query(frame, classes, confidence_threshold):
+    image = Image.fromarray(frame)
     inputs = processor(images=image, text=classes, return_tensors="pt").to(device)
     with torch.no_grad():
         start = time.time()
@@ -124,7 +118,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         """
     )
     gr.Markdown(
-        "Simply upload a video, and write the objects you want to detect! You can also play with confidence threshold, image size, or try the examples below. 👇"
+        "Simply upload a video, and write the objects you want to detect! You can also play with confidence threshold or try the examples below. 👇"
     )
 
     with gr.Row():
@@ -147,29 +141,23 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             value=0.2,
             step=0.05,
         )
-        max_side = gr.Slider(
-            label="Image Size",
-            minimum=240,
-            maximum=1080,
-            value=640,
-            step=10,
-        )
+
     with gr.Row():
         submit = gr.Button(variant="primary")
 
     example = gr.Examples(
         examples=[
-            ["./football.mp4", 0.3, "person, ball, shoe", 640],
-            ["./cat.mp4", 0.2, "cat", 640],
-            ["./safari2.mp4", 0.3, "elephant, giraffe, springbok, zebra", 640],
+            ["./football.mp4", 0.3, "person, ball, shoe"],
+            ["./cat.mp4", 0.2, "cat"],
+            ["./safari2.mp4", 0.3, "elephant, giraffe, springbok, zebra"],
         ],
-        inputs=[input_video, conf, classes, max_side],
+        inputs=[input_video, conf, classes],
         outputs=output_video,
     )
 
     submit.click(
         fn=process_video,
-        inputs=[input_video, conf, classes, max_side],
+        inputs=[input_video, conf, classes],
         outputs=[output_video, actual_fps],
     )