updated app.py

app.py

@@ -41,12 +41,16 @@ def visualize(image, results, box_color=(0, 255, 0), text_color=(0, 0, 255)):
     return output
 
 def detect_faces(input_image):
+    input_image = cv.cvtColor(input_image, cv.COLOR_RGB2BGR)
     h, w, _ = input_image.shape
     model.setInputSize([w, h])
     results = model.infer(input_image)
     if results is None or len(results) == 0:
+        input_image = cv.cvtColor(input_image, cv.COLOR_BGR2RGB)
         return input_image
-    return visualize(input_image, results)
+    output = visualize(input_image, results)
+    output = cv.cvtColor(output, cv.COLOR_BGR2RGB)
+    return output
 
 # Gradio Interface
 demo = gr.Interface(