KoonJamesZ commited on
Commit
6f72ed8
·
verified ·
1 Parent(s): dfa6fd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ def extract_audio(video_path):
89
  return ""
90
 
91
  output_dir = '/content/images'
92
- model_yolo = YOLO('/content/drive/MyDrive/CCIB-AI-YOLO/runs/detect/train/weights/best.pt')
93
 
94
  def extract_top_weapon_frames(video_path, threshold=30):
95
  os.makedirs(output_dir, exist_ok=True)
 
89
  return ""
90
 
91
  output_dir = '/content/images'
92
+ model_yolo = YOLO('/model/best.pt')
93
 
94
  def extract_top_weapon_frames(video_path, threshold=30):
95
  os.makedirs(output_dir, exist_ok=True)