gk8686 commited on
Commit
54c5767
1 Parent(s): 01e234e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,7 +29,7 @@ for i, url in enumerate(file_urls):
29
  f"image_{i}.jpg"
30
  )
31
 
32
- #model = YOLO('best.pt')
33
  path = [['image_0.jpg'], ['image_1.jpg']]
34
  video_path = [['video.mp4']]
35
 
@@ -76,7 +76,7 @@ def show_preds_video(video_path):
76
  frame_copy,
77
  (int(det[0]), int(det[1])),
78
  (int(det[2]), int(det[3])),
79
- #color=(0, 0, 255),
80
  thickness=2,
81
  lineType=cv2.LINE_AA
82
  )
 
29
  f"image_{i}.jpg"
30
  )
31
 
32
+ model = YOLO('best.pt')
33
  path = [['image_0.jpg'], ['image_1.jpg']]
34
  video_path = [['video.mp4']]
35
 
 
76
  frame_copy,
77
  (int(det[0]), int(det[1])),
78
  (int(det[2]), int(det[3])),
79
+ color=(0, 0, 255),
80
  thickness=2,
81
  lineType=cv2.LINE_AA
82
  )