add face detection

app.py

@@ -1,14 +1,34 @@
 import gradio as gr
+# load libraries
+from huggingface_hub import hf_hub_download
+from ultralytics import YOLO
+from supervision import Detections
+from PIL import Image
 
-def greet(name):
-    return "Codingon " + name + "!"
+# download model
+model_path = hf_hub_download(
+    repo_id="arnabdhar/YOLOv8-Face-Detection", filename="model.pt"
+)
 
-with gr.Blocks() as demo:
-    name = gr.Textbox(label="Name")
-    output = gr.Textbox(label="Output Box")
-    greet_btn = gr.Button("Greet")
-    greet_btn.click(fn=greet, inputs=name, outputs=output, api_name="greet")
+# load model
+model = YOLO(model_path)
 
+import cv2
+import numpy as np
+import gradio as gr
+
+def sepia(input_img):
+    output = model(input_img)
+    results = Detections.from_ultralytics(output[0])
+    arr_int = results.xyxy.astype(int)
+
+    for x, y, x2, y2 in arr_int:
+      cv2.rectangle(input_img, (x, y), (x2, y2), (0, 255, 0), 2)
+
+    return input_img
+
+demo = gr.Interface(sepia, gr.Image(), "image")
 if __name__ == "__main__":
     demo.launch()
-    
+
+