jgyasu's picture
Add entire pipeline
060ac52
raw
history blame contribute delete
534 Bytes
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()