Fix: argument mismatch

app.py

@@ -8,6 +8,7 @@ from yolov5 import xai_yolov5
 from yolov8 import xai_yolov8s
 
 def process_image(image, yolo_versions=["yolov5"]):
+    # Convert image from PIL to NumPy array
     image = np.array(image)
     image = cv2.resize(image, (640, 640))
 
@@ -21,6 +22,7 @@ def process_image(image, yolo_versions=["yolov5"]):
             result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
     return result_images
 
+
 interface = gr.Interface(
     fn=process_image,
     inputs=[
@@ -41,5 +43,6 @@ interface = gr.Interface(
     live=True  
 )
 
+
 if __name__ == "__main__":
     interface.launch()