Hunyuan-Avatar / app.py
rahul7star's picture
Create app.py
c1f7300 verified
raw
history blame
551 Bytes
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()