import gradio as gr | |
from UI.gradio import create_gradio_interface | |
from pathlib import Path | |
from utils.config import load_config | |
project_root = Path(__file__).resolve().parent | |
config_path = project_root / "utils" / "config.yaml" | |
config = load_config(config_path)['PECCAVI_TEXT'] | |
def main(): | |
""" | |
This function is the entry point for the PECCAVI Watermarking Model. | |
It creates the Gradio interface for the model and runs it. | |
""" | |
create_gradio_interface(config).launch() | |
if __name__ == "__main__": | |
main() |