kisa-misa commited on
Commit
5437fbc
·
1 Parent(s): c6c2ece

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -6,11 +6,9 @@ from glob import glob
6
  import gradio as gr
7
  #model = Yolov4(weight_path="yolov4.weights", class_name_path='coco_classes.txt')
8
  def gradio_wrapper(img):
9
- %cd %cd /content/YOLOv8-DeepSORT-Object-Tracking-and-Pulse-Estimation-2/ultralytics/yolo/v8/detect
10
  !python predict.py model=yolov8x6.pt source=img
11
  #print(np.shape(img))
12
- results = model.predict(img)
13
- return results[0]
14
  demo = gr.Interface(
15
  gradio_wrapper,
16
  #gr.Image(source="webcam", streaming=True, flip=True),
 
6
  import gradio as gr
7
  #model = Yolov4(weight_path="yolov4.weights", class_name_path='coco_classes.txt')
8
  def gradio_wrapper(img):
9
+ %cd ultralytics/yolo/v8/detect/
10
  !python predict.py model=yolov8x6.pt source=img
11
  #print(np.shape(img))
 
 
12
  demo = gr.Interface(
13
  gradio_wrapper,
14
  #gr.Image(source="webcam", streaming=True, flip=True),