KoonJamesZ commited on
Commit
1fe80a0
·
verified ·
1 Parent(s): 1df3240

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('/model/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('/home/user/app/model/best.pt')
93
 
94
  def extract_top_weapon_frames(video_path, threshold=30):
95
  os.makedirs(output_dir, exist_ok=True)