NKASG commited on
Commit
ea8a726
·
verified ·
1 Parent(s): dae5238

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -29,7 +29,7 @@ for i, url in enumerate(file_urls):
29
  f"image_{i}.jpg"
30
  )
31
 
32
- model = YOLO('best20.pt')
33
  path = [['image_0.jpg'], ['image_1.jpg']]
34
  video_path = [['video.mp4']]
35
 
@@ -83,11 +83,11 @@ def show_preds_video(video_path):
83
  yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)
84
 
85
 
86
- inputs_video = [
87
  gr.components.Video(type="filepath", label="Input Video"),
88
 
89
  ]
90
- outputs_video = [
91
  gr.components.Image(type="numpy", label="Output Image"),
92
  ]
93
  interface_video = gr.Interface(
 
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
 
 
83
  yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)
84
 
85
 
86
+ inputs_video = [
87
  gr.components.Video(type="filepath", label="Input Video"),
88
 
89
  ]
90
+ outputs_video = [
91
  gr.components.Image(type="numpy", label="Output Image"),
92
  ]
93
  interface_video = gr.Interface(