Update app.py

app.py

@@ -1,32 +1,16 @@
-# import gradio as gr
-# import torch
-
-# model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
-# # Define the face detector function
-# def detect_faces(image):
-#    # Loading in yolov5s - you can switch to larger models such as yolov5m or yolov5l, or smaller such as yolov5n
-#     results = model(image)
-
-#     return results.render()[0]
-
-# # Create a Gradio interface
-# iface = gr.Interface(fn=detect_faces, inputs="image", outputs="image")
-
-# # Launch the interface
-# iface.launch(debug=True)
-
 import gradio as gr
+import torch
 
+model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
+# Define the face detector function
+def detect_faces(image):
+   # Loading in yolov5s - you can switch to larger models such as yolov5m or yolov5l, or smaller such as yolov5n
+    results = model(image)
 
-def snap(image, video):
-    return [image, video]
-
+    return results.render()[0]
 
-demo = gr.Interface(
-    snap,
-    [gr.Image(source="webcam", tool=None), gr.Video(source="webcam")],
-    ["image", "video"],
-)
+# Create a Gradio interface
+iface = gr.Interface(fn=detect_faces, inputs=gr.Video(source="webcam"), outputs="image")
 
-if __name__ == "__main__":
-    demo.launch()
+# Launch the interface
+iface.launch(debug=True)