deep-detect-api / model.py
muhammadusmanalyy's picture
Backend deployment
6ab5c33 verified
Raw
History Blame Contribute Delete
312 Bytes
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)