alpha

README.md

@@ -4,9 +4,9 @@ emoji: 🏢
 colorFrom: yellow
 colorTo: green
 sdk: gradio
-sdk_version: 5.17.1
+sdk_version: 4.12.0
 app_file: app.py
-pinned: false
+pinned: true
 short_description: Image-to-Video
 ---
 

app.py

@@ -0,0 +1,48 @@
+# app.py
+import gradio as gr
+from hunyuan_model import HunyuanVideoPipeline
+from diffusers.utils import export_to_video
+import huggingface_hub
+import spaces
+
+@spaces.GPU(duration=120)
+def generate_video(image, video_length=85, infer_steps=30, seed=123):
+    pipeline = HunyuanVideoPipeline.from_pretrained(
+        "AeroScripts/leapfusion-hunyuan-image2video",
+        variant="fp8",
+        torch_dtype=torch.float16
+    )
+
+    generator = torch.Generator().manual_seed(seed)
+    frames = pipeline(
+        image,
+        video_length=video_length,
+        num_inference_steps=infer_steps,
+        generator=generator
+    ).frames
+
+    video_path = export_to_video(frames)
+    return video_path
+
+interface = gr.Blocks(title="Hunyuan-ITV")
+
+with interface:
+    gr.Markdown("# Hunyuan Image-to-Video Converter")
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(label="Input Image", type="filepath")
+            video_length = gr.Slider(16, 120, value=85, label="Frame Count")
+            infer_steps = gr.Slider(10, 50, value=30, step=5, label="Inference Steps")
+            seed = gr.Number(123, label="Random Seed")
+            submit_btn = gr.Button("Generate Video")
+        with gr.Column():
+            output_video = gr.Video(label="Generated Video")
+
+    submit_btn.click(
+        generate_video,
+        inputs=[input_image, video_length, infer_steps, seed],
+        outputs=output_video
+    )
+
+if __name__ == "__main__":
+    interface.launch(server_name="0.0.0.0", server_port=7860)

cache_manager.py

@@ -0,0 +1,7 @@
+from diskcache import Cache
+
+video_cache = Cache("/tmp/hunyuan_videos")
+
+@video_cache.memoize(expire=3600)
+def cached_generation(image, params):
+    return generate_video(image, **params)

config.json

@@ -0,0 +1,6 @@
+{
+  "HUNYUAN_MODEL": "AeroScripts/leapfusion-hunyuan-image2video",
+  "MAX_FRAME_SIZE": 960,
+  "QUANTIZATION_MODE": "fp8",
+  "SAFETY_CHECKER": "disabled"
+}

requirements.txt

@@ -0,0 +1,8 @@
+# requirements.txt
+gradio==4.12.0
+torch==2.3.1
+accelerate==0.30.0
+transformers==4.40.0
+diffusers==0.28.0
+safetensors==0.4.2
+huggingface_hub