Update app.py

app.py

@@ -28,7 +28,10 @@ def count_unique_trucks(video_path):
     unique_truck_ids = set()
     truck_history = {}
 
-    frame_skip = 5  # Process every 5th frame for efficiency
+    # Get FPS of the video
+    fps = int(cap.get(cv2.CAP_PROP_FPS))
+    frame_skip = fps * 7  # Skip frames every 5 seconds
+
     frame_count = 0
 
     while True:
@@ -38,7 +41,7 @@ def count_unique_trucks(video_path):
 
         frame_count += 1
         if frame_count % frame_skip != 0:
-            continue  # Skip frames to improve efficiency
+            continue  # Skip frames to process only every 5 seconds
 
         # Run YOLOv12x inference
         results = model(frame, verbose=False)
@@ -84,22 +87,3 @@ def count_unique_trucks(video_path):
     cap.release()
 
     return {"Total Unique Trucks": len(unique_truck_ids)}
-
-# Gradio UI function
-def analyze_video(video_file):
-    result = count_unique_trucks(video_file)
-    return "\n".join([f"{key}: {value}" for key, value in result.items()])
-
-# Define Gradio interface
-import gradio as gr
-iface = gr.Interface(
-    fn=analyze_video,
-    inputs=gr.Video(label="Upload Video"),
-    outputs=gr.Textbox(label="Analysis Result"),
-    title="YOLOv12x Unique Truck Counter",
-    description="Upload a video to count unique trucks using YOLOv12x and SORT tracking."
-)
-
-# Launch the Gradio app
-if __name__ == "__main__":
-    iface.launch()