Update app.py

app.py

@@ -1,23 +1,36 @@
-import gradio as gr
-import torch
-
-def snap(video):
-    return [video]
+# import gradio as gr
+# import torch
 
-model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
-# Define the face detector function
+# 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(
+# # 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
+
+def snap(video):
+    results = model(image)
+    return [results.render()[0], video]
+    
+# 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]
+
+demo = gr.Interface(
     snap,
-    [gr.Video(source="webcam")],
-    ["video"],
+    inputs = gr.Video(source="webcam"),
+    outputs = "image",
 )
 
-# Launch the interface
-iface.launch(debug=True)
+if __name__ == "__main__":
+    demo.launch()