Update app.py

app.py

@@ -7,11 +7,24 @@ from diffusers.utils import export_to_video
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 import logging
+import importlib.util
 
 # Thiết lập logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Kiểm tra backend
+def check_backend():
+    if importlib.util.find_spec("imageio") and importlib.util.find_spec("imageio_ffmpeg"):
+        logger.info("imageio and imageio-ffmpeg are installed. Using recommended backend.")
+    elif importlib.util.find_spec("cv2"):
+        logger.warning("imageio/imageio-ffmpeg not found. Using legacy OpenCV backend.")
+    else:
+        logger.error("No suitable backend found for export_to_video. Please install imageio and imageio-ffmpeg.")
+        raise ImportError("Missing backend for export_to_video. Install with: pip install imageio imageio-ffmpeg")
+
+check_backend()
+
 # Tạo thư mục lưu video trong không gian làm việc của Spaces
 output_dir = "/home/user/app/outputs"
 os.makedirs(output_dir, exist_ok=True)
@@ -92,7 +105,7 @@ def generate_image(prompt, base="Realistic", motion="", step=1, progress=gr.Prog
         name = str(uuid.uuid4()).replace("-", "")
         path = os.path.join(output_dir, f"{name}.mp4")
         logger.info(f"Saving video to {path}")
-        export_to_video(output.frames[0], path, fps=10)  # Loại bỏ codec
+        export_to_video(output.frames[0], path, fps=10)
 
         # Kiểm tra file video
         if not os.path.exists(path):