rosebe commited on
Commit
da8e3ad
·
1 Parent(s): 2e0ea1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -6,11 +6,11 @@ def snap(video):
6
 
7
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
8
  # Define the face detector function
9
- def detect_faces(image):
10
- # Loading in yolov5s - you can switch to larger models such as yolov5m or yolov5l, or smaller such as yolov5n
11
- results = model(image)
12
 
13
- return results.render()[0]
14
 
15
  # Create a Gradio interface
16
  iface = gr.Interface(
 
6
 
7
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
8
  # Define the face detector function
9
+ # def detect_faces(image):
10
+ # # Loading in yolov5s - you can switch to larger models such as yolov5m or yolov5l, or smaller such as yolov5n
11
+ # results = model(image)
12
 
13
+ # return results.render()[0]
14
 
15
  # Create a Gradio interface
16
  iface = gr.Interface(