import os
from huggingface_hub import snapshot_download

MODEL_DIR = "./weights/ckpts/hunyuan-video-t2v-720p"
CHECKPOINT_PATH = os.path.join(MODEL_DIR, "transformers", "mp_rank_00_model_states.pt")

if not os.path.exists(CHECKPOINT_PATH):
    snapshot_download(
        repo_id="tencent/HunyuanVideo-Avatar",
        local_dir=MODEL_DIR,
        local_dir_use_symlinks=False
    )

# Now import and launch Gradio UI
import gradio as gr
def run_app(input):
    return "Model is loaded!"
gr.Interface(fn=run_app, inputs="text", outputs="text").launch()