Spaces:
Starting
on
L4
Starting
on
L4
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() | |