import gradio as gr
from main import build_ui

# Build Gradio UI
demo = build_ui()