Update app.py

app.py

@@ -7,7 +7,7 @@ from PIL import Image
 import numpy as np
 from transformers import pipeline
 
-pipe = pipeline("image-classification", model="NTQAI/pedestrian_gender_recognition")
+#pipe = pipeline("image-classification", model="NTQAI/pedestrian_gender_recognition")
 model_path = hf_hub_download(repo_id="arnabdhar/YOLOv8-Face-Detection", filename="model.pt")
 model = YOLO(model_path)
 
@@ -26,14 +26,14 @@ def detect_faces(image):
     for (x, y, w, h) in faces:
         cv2.rectangle(image_np, (x, y), (x+w, y+h), (0, 255, 0), 2)
     
-    label_out = pipe(image)
+    #label_out = pipe(image)
     
-    return (image_np,im,label_out[0]['label'])
+    return (image_np,im)#,label_out[0]['label'])
 
 interface = gr.Interface(
     fn=detect_faces,
     inputs=gr.Image(label='Upload Image'),
-    outputs=[gr.Image(label='Original'),gr.Image(label='Deep learning'),'text'],
+    outputs=[gr.Image(label='Original'),gr.Image(label='Deep learning')],
     title="Face Detection Deep Learning",
     description="Upload an image, and the model will detect faces and draw bounding boxes around them.",
 )