Update app.py

app.py

@@ -24,7 +24,7 @@ line_zone_annotator = sv.LineZoneAnnotator(
 )
 
 
-def process_video(video_path, orientation="Ngang", frame_skip=1):
+def process_video(video_path, orientation="Ngang"):
     """Xử lý video: nhận dạng phương tiện và trả về video đã annotate."""
     if video_path is None:
         return None
@@ -59,17 +59,9 @@ def process_video(video_path, orientation="Ngang", frame_skip=1):
     previous_positions = {}
     class_counts = {name: 0 for name in SELECTED_CLASS_NAMES}
     crossed_ids = set()
-    last_annotated_frame = None
 
     def callback(frame: np.ndarray, index: int) -> np.ndarray:
-        nonlocal previous_positions, class_counts, crossed_ids, last_annotated_frame
-
-        try:
-            fs = max(1, int(frame_skip))
-        except Exception:
-            fs = 1
-        if fs > 1 and index % fs != 0:
-            return last_annotated_frame if last_annotated_frame is not None else frame
+        nonlocal previous_positions, class_counts, crossed_ids
 
         results = model(frame, verbose=False)[0]
         detections = sv.Detections.from_ultralytics(results)
@@ -128,8 +120,6 @@ def process_video(video_path, orientation="Ngang", frame_skip=1):
                         (x0 + 10, y0 + 60 + i * 28),
                         cv2.FONT_HERSHEY_SIMPLEX, 0.7, (255, 255, 255), 2)
 
-        last_annotated_frame = annotator_frame
-
         return annotator_frame
 
     output_path = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False).name
@@ -155,23 +145,11 @@ with gr.Blocks(title="Object Detection", theme=gr.themes.Soft()) as demo:
         label="Vị trí line"
     )
 
-    frame_skip_input = gr.Slider(
-        minimum=1,
-        maximum=10,
-        step=1,
-        value=1,
-        label="Frame skip (1 = xử lý mọi frame)"
-    )
-
     btn = gr.Button("▶️ Xử lý video")
-    btn.click(
-        fn=process_video,
-        inputs=[video_input, orientation_input, frame_skip_input],
-        outputs=video_output
-    )
+    btn.click(fn=process_video, inputs=[video_input, orientation_input], outputs=video_output)
 
     gr.Markdown("""
-    ### Sinh viên: Trần Hải Nam - 223332840
+    ### Author: Trần Hải Nam - 223332840
     """)
 
 if __name__ == "__main__":