rosebe commited on
Commit
1f86d79
·
1 Parent(s): 9b4be67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,7 +19,7 @@ import gradio as gr
19
 
20
  def snap(video):
21
  results = model(image)
22
- return [results.render()[0], video]
23
 
24
  # def detect_faces(image):
25
  # # # Loading in yolov5s - you can switch to larger models such as yolov5m or yolov5l, or smaller such as yolov5n
@@ -28,8 +28,8 @@ def snap(video):
28
 
29
  demo = gr.Interface(
30
  snap,
31
- inputs = gr.Video(source="webcam"),
32
- outputs = "image",
33
  )
34
 
35
  if __name__ == "__main__":
 
19
 
20
  def snap(video):
21
  results = model(image)
22
+ return [results.render()[0]]
23
 
24
  # def detect_faces(image):
25
  # # # Loading in yolov5s - you can switch to larger models such as yolov5m or yolov5l, or smaller such as yolov5n
 
28
 
29
  demo = gr.Interface(
30
  snap,
31
+ gr.Video(source="webcam"),
32
+ "image",
33
  )
34
 
35
  if __name__ == "__main__":