Update app.py

app.py

@@ -1,9 +1,25 @@
+from huggingface_hub import hf_hub_download
+from ultralytics import YOLO
 import cv2
 from PIL import Image
+import gradio as gr
 
-img = "https://farm5.staticflickr.com/4139/4887614566_6b57ec4422_z.jpg"
-output = model(img)
-pred = output[0].plot()
-pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
-pred = Image.fromarray(pred)
-pred
+def detect_objects(img):
+    # Download the model
+    path = hf_hub_download("Bingsu/adetailer", "face_yolov8n.pt")
+    model = YOLO(path)
+
+    # Run the model
+    output = model(img)
+    pred = output[0].plot()
+    pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
+    pred = Image.fromarray(pred)
+
+    # Return the output
+    return pred
+
+# Define a Gradio interface for the function
+iface = gr.Interface(fn=detect_objects, inputs="image", outputs="image")
+
+# Launch the interface
+iface.launch()