Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 import supervision as sv
 from ultralytics import YOLO
 
-model = YOLO("yolo11n.pt")
+model = YOLO("yolov8n.pt")
 CLASS_NAMES_DICT = model.model.names
 
 SELECTED_CLASS_NAMES = ['car', 'bus', 'truck', 'motorcycle']
@@ -35,7 +35,8 @@ def process_video(video_path, orientation="Horizontal"):
     video_info = sv.VideoInfo.from_video_path(video_path)
     w, h = video_info.width, video_info.height
 
-    is_horizontal = orientation.lower().startswith("n")
+    ori = str(orientation).strip().lower()
+    is_horizontal = ori.startswith("n") or ori.startswith("h")
     if is_horizontal:
         line_pos = int(h * 0.5)
         line_start = sv.Point(int(w * 0.005), line_pos)