Initial commit

README.md

@@ -1,14 +1,20 @@
----
-title: ImageToVideo
-emoji: 👁
-colorFrom: pink
-colorTo: green
-sdk: gradio
-sdk_version: 5.25.2
-app_file: app.py
-pinned: false
-license: apache-2.0
-short_description: image to video generator
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Ứng dụng tạo video từ hình ảnh
+
+Ứng dụng này cho phép người dùng tải lên hình ảnh và tạo video chuyển động từ hình ảnh đó sử dụng mô hình AI.
+
+## Cách sử dụng
+
+1. Tải lên một hình ảnh
+2. Nhập mô tả cho chuyển động bạn muốn tạo
+3. Điều chỉnh các tham số (tùy chọn)
+4. Nhấn "Tạo video"
+5. Đợi quá trình xử lý hoàn tất và tải xuống video
+
+## Mô hình được sử dụng
+
+Ứng dụng này sử dụng mô hình AnimateDiff để tạo hoạt ảnh từ hình ảnh tĩnh.
+
+## Tham khảo
+
+- [AnimateDiff GitHub](https://github.com/guoyww/AnimateDiff)
+- [HuggingFace Diffusers](https://huggingface.co/docs/diffusers/index)

app.py

@@ -0,0 +1,94 @@
+import gradio as gr
+import torch
+from diffusers import AnimationPipeline, DDIMScheduler
+from huggingface_hub import hf_hub_download
+import os
+from PIL import Image
+import numpy as np
+from moviepy.editor import ImageSequenceClip
+
+# Kiểm tra xem có GPU không
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"Using device: {device}")
+
+# Tải mô hình (đây là ví dụ với AnimateDiff)
+def load_model():
+    pipe = AnimationPipeline.from_pretrained(
+        "guoyww/animatediff-motion-adapter-v1-5",
+        torch_dtype=torch.float16 if device == "cuda" else torch.float32
+    )
+    pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
+    pipe = pipe.to(device)
+    return pipe
+
+# Hàm tạo video từ hình ảnh
+def generate_video(image, prompt, num_frames=24, guidance_scale=7.5):
+    # Tiền xử lý hình ảnh
+    if image is None:
+        return None, "Vui lòng tải lên một hình ảnh"
+    
+    # Đảm bảo hình ảnh là định dạng RGB
+    if image.mode != "RGB":
+        image = image.convert("RGB")
+    
+    # Thay đổi kích thước hình ảnh về 512x512 (kích thước phổ biến cho mô hình)
+    image = image.resize((512, 512))
+    
+    try:
+        # Tải mô hình (có thể mất thời gian lần đầu tiên)
+        pipe = load_model()
+        
+        # Tạo video
+        output = pipe(
+            prompt=prompt,
+            image=image,
+            num_inference_steps=30,
+            guidance_scale=guidance_scale,
+            num_frames=num_frames
+        )
+        
+        # Xử lý kết quả
+        frames = output.frames[0]
+        
+        # Chuyển đổi frames thành video
+        frame_array = [np.array(frame) for frame in frames]
+        clip = ImageSequenceClip(frame_array, fps=8)
+        
+        # Lưu video
+        output_path = "output_video.mp4"
+        clip.write_videofile(output_path, codec="libx264")
+        
+        return output_path, "Video đã được tạo thành công!"
+    except Exception as e:
+        return None, f"Lỗi: {str(e)}"
+
+# Tạo giao diện Gradio
+with gr.Blocks(title="Ứng dụng tạo video từ hình ảnh") as demo:
+    gr.Markdown("# Tạo video từ hình ảnh sử dụng AI")
+    gr.Markdown("Tải lên một hình ảnh và nhập mô tả để tạo video chuyển động.")
+    
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="pil", label="Tải lên hình ảnh")
+            prompt_input = gr.Textbox(label="Mô tả chuyển động", 
+                                     placeholder="Mô tả cách bạn muốn hình ảnh chuyển động...",
+                                     value="A gentle camera zoom in, cinematic lighting")
+            num_frames = gr.Slider(minimum=16, maximum=48, value=24, step=8, label="Số khung hình")
+            guidance = gr.Slider(minimum=1.0, maximum=15.0, value=7.5, step=0.5, label="Guidance Scale")
+            submit_btn = gr.Button("Tạo video")
+        
+        with gr.Column():
+            output_video = gr.Video(label="Video kết quả")
+            output_message = gr.Textbox(label="Thông báo")
+    
+    submit_btn.click(
+        fn=generate_video,
+        inputs=[image_input, prompt_input, num_frames, guidance],
+        outputs=[output_video, output_message]
+    )
+    
+    gr.Markdown("### Lưu ý")
+    gr.Markdown("- Quá trình tạo video có thể mất vài phút, đặc biệt là khi tải mô hình lần đầu")
+    gr.Markdown("- Kích thước hình ảnh sẽ được thay đổi về 512x512 pixel")
+
+demo.launch()

requirements.txt

@@ -0,0 +1,9 @@
+gradio==4.0.2
+torch
+torchvision
+diffusers
+transformers
+accelerate
+huggingface_hub
+pillow
+moviepy