viswanani commited on
Commit
e4e5e23
·
verified ·
1 Parent(s): 660dca8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  os.environ['YOLO_CONFIG_DIR'] = '/tmp'
9
 
10
  # Initialize the YOLO model
11
- model = YOLO('yolov8n.pt') # Ensure yolov8n.pt is in the same directory as this script
12
 
13
  def process_video(video):
14
  print("Processing video...")
 
8
  os.environ['YOLO_CONFIG_DIR'] = '/tmp'
9
 
10
  # Initialize the YOLO model
11
+ model = YOLO('yolov5n.pt') # Ensure yolov5n.pt is in the same directory as this script
12
 
13
  def process_video(video):
14
  print("Processing video...")