awacke1 commited on
Commit
97b681a
1 Parent(s): 58b5407

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -4,15 +4,12 @@ import numpy as np
4
  from glob import glob
5
  from models import Yolov4
6
  import gradio as gr
7
-
8
  model = Yolov4(weight_path="yolov4.weights", class_name_path='coco_classes.txt')
9
-
10
  def gradio_wrapper(img):
11
  global model
12
  print(np.shape(img))
13
  results = model.predict(img)
14
  return results[0]
15
-
16
  demo = gr.Interface(
17
  gradio_wrapper,
18
  gr.Image(source="webcam", streaming=True, flip=True),
 
4
  from glob import glob
5
  from models import Yolov4
6
  import gradio as gr
 
7
  model = Yolov4(weight_path="yolov4.weights", class_name_path='coco_classes.txt')
 
8
  def gradio_wrapper(img):
9
  global model
10
  print(np.shape(img))
11
  results = model.predict(img)
12
  return results[0]
 
13
  demo = gr.Interface(
14
  gradio_wrapper,
15
  gr.Image(source="webcam", streaming=True, flip=True),