ntsc207 commited on
Commit
5abe39b
·
verified ·
1 Parent(s): c6ede3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,6 +4,7 @@ from detect_deepsort import run_deepsort
4
  from detect_strongsort import run_strongsort
5
  from detect import run
6
  import os
 
7
  from PIL import Image
8
  import cv2
9
  import numpy as np
@@ -110,9 +111,9 @@ def app():
110
  ],
111
  value="None"
112
  )
 
113
  gr.Examples(['./img_examples/Exam_1.png','./img_examples/Exam_2.png','./img_examples/Exam_3.png','./img_examples/Exam_4.png','./img_examples/Exam_5.png'], inputs=img_path,label = "Image Example", cache_examples = False)
114
  gr.Examples(['./video_examples/video_1.mp4', './video_examples/video_2.mp4','./video_examples/video_3.mp4','./video_examples/video_4.mp4','./video_examples/video_5.mp4'], inputs=vid_path, label = "Video Example", cache_examples = False)
115
- yolov9_infer = gr.Button(value="Inference")
116
  with gr.Column():
117
  gr.HTML("<h2>Output</h2>")
118
  output_image = gr.Image(type="numpy",label="Output")
 
4
  from detect_strongsort import run_strongsort
5
  from detect import run
6
  import os
7
+ import torch
8
  from PIL import Image
9
  import cv2
10
  import numpy as np
 
111
  ],
112
  value="None"
113
  )
114
+ yolov9_infer = gr.Button(value="Inference")
115
  gr.Examples(['./img_examples/Exam_1.png','./img_examples/Exam_2.png','./img_examples/Exam_3.png','./img_examples/Exam_4.png','./img_examples/Exam_5.png'], inputs=img_path,label = "Image Example", cache_examples = False)
116
  gr.Examples(['./video_examples/video_1.mp4', './video_examples/video_2.mp4','./video_examples/video_3.mp4','./video_examples/video_4.mp4','./video_examples/video_5.mp4'], inputs=vid_path, label = "Video Example", cache_examples = False)
 
117
  with gr.Column():
118
  gr.HTML("<h2>Output</h2>")
119
  output_image = gr.Image(type="numpy",label="Output")