retest

test.py

@@ -1,21 +1,19 @@
-%pip install -r requirements.txt
-%pip install gradio
-
 import gradio as gr
-import numpy as np
-
-def sepia(input_img):
-    sepia_filter - np.array([
-        [0.393, 0.769, 0.189],
-        [0.349, 0.686, 0.168],
-        [0.272, 0.534, 0.131]
-    ])
-    sepia_img = input_img.dot(sepia_filter.T)
-    sepia_img /= sepia_img.max()
-    return sepia_img
-
-demo = gr.Interface(sepia, gr.Image(), "Image")
-
+from huggingface_hub import hf_hub_download
+from ultralytics import YOLO
+from supervision import Detections
+from PIL import Image
+import cv2
+model_path = hf_hub_download(repo_id="arnabdhar/YOLOv8-Face-Detection", filename="model.pt")
+model = YOLO(model_path)
+def greet(img):
+    output = model(img)
+    results = Detections.from_ultralytics(output[0])
+    arr_int = results.xyxy.astype(int)
+    for x, y, x2, y2 in arr_int:
+        cv2.rectangle(img, (x, y), (x2, y2), (0, 255, 0), 2)
+    return img
+demo = gr.Interface(fn=greet, inputs="image", outputs="image")
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(show_api=False, share=True)