Update app.py

app.py

@@ -3,33 +3,33 @@ from mtcnn.mtcnn import MTCNN
 import cv2
 import numpy as np
 
-# Function to perform face detection using MTCNN
+# Function to detect faces using MTCNN
 def detect_faces(image):
-    # Load the image
-    img = cv2.imdecode(np.frombuffer(image.read(), np.uint8), -1)
+    # Convert image to RGB format
+    image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
-    # Create an MTCNN detector
+    # Detect faces using MTCNN
     detector = MTCNN()
+    faces = detector.detect_faces(image_rgb)
 
-    # Detect faces in the image
-    faces = detector.detect_faces(img)
-
-    # Draw bounding boxes around the faces
+    # Draw bounding boxes around detected faces
     for face in faces:
         x, y, width, height = face['box']
-        cv2.rectangle(img, (x, y), (x + width, y + height), (0, 255, 0), 2)
-
-    # Convert the image to RGB format for Gradio
-    img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+        cv2.rectangle(image, (x, y), (x + width, y + height), (255, 0, 0), 2)
 
-    return img_rgb
+    return image
 
-# Create a Gradio interface
+# Gradio Interface
 iface = gr.Interface(
     fn=detect_faces,
-    inputs=gr.Image(),
-    outputs="image"
+    inputs=gr.Image(type="numpy", label="Input Image"),
+    outputs=gr.Image(type="numpy", label="Output Image"),
+    live=True,
+    capture_session=True,
+    interpretation="default",
+    title="MTCNN Face Detection",
+    description="Detect faces in an image using MTCNN.",
 )
 
-# Launch the Gradio interface
+# Launch the Gradio Interface
 iface.launch()