model.py

from inference import predict_label

def predict(image_path: str) -> str:
    """

    Predict if the image is real or Deep-Fake.



    Args:

        image_path (str): Absolute or relative path to the image file.



    Returns:

        str: "ai" or "real".

    """
    return predict_label(image_path)