Update app.py

app.py

@@ -12,8 +12,8 @@ import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Tạo thư mục lưu video
-output_dir = "./outputs"
+# 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)
 
 # Constants
@@ -62,11 +62,15 @@ def generate_image(prompt, base="Realistic", motion="", step=1, progress=gr.Prog
 
         # Tải motion LoRA (tùy chọn)
         if motion_loaded != motion:
-            pipe.unload_lora_weights()
-            if motion != "":
-                pipe.load_lora_weights(motion, adapter_name="motion")
-                pipe.set_adapters(["motion"], [0.7])
-            motion_loaded = motion
+            try:
+                pipe.unload_lora_weights()
+                if motion != "":
+                    pipe.load_lora_weights(motion, adapter_name="motion")
+                    pipe.set_adapters(["motion"], [0.7])
+                motion_loaded = motion
+            except Exception as e:
+                logger.warning(f"Failed to load LoRA weights: {str(e)}. Proceeding without LoRA.")
+                motion_loaded = ""
 
         progress((0, step))
         def progress_callback(i, t, z):
@@ -79,16 +83,18 @@ def generate_image(prompt, base="Realistic", motion="", step=1, progress=gr.Prog
                 guidance_scale=1.2,
                 num_inference_steps=step,
                 callback=progress_callback,
-                callback_steps=1
+                callback_steps=1,
+                width=256,  # Giảm độ phân giải để tăng tốc
+                height=256
             )
 
-        # Xuất video
+        # Xuất video với codec H.264
         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)
+        export_to_video(output.frames[0], path, fps=10, codec="h264")
 
-        # Kiểm tra xem file video có tồn tại không
+        # Kiểm tra file video
         if not os.path.exists(path):
             raise FileNotFoundError(f"Video file {path} was not created.")
         
@@ -145,7 +151,7 @@ with gr.Blocks(css=css) as demo:
                     ("Roll left", "guoyww/animatediff-motion-lora-rolling-anticlockwise"),
                     ("Roll right", "guoyww/animatediff-motion-lora-rolling-clockwise"),
                 ],
-                value="guoyww/animatediff-motion-lora-zoom-in",
+                value="",
                 interactive=True,
                 elem_id="gr-input"
             )
@@ -158,7 +164,7 @@ with gr.Blocks(css=css) as demo:
             )
             submit = gr.Button("Generate Video", variant="primary", elem_id="gr-button")
         with gr.Column(elem_id="output"):
-            video = gr.Video(label="AnimateDiff-Lightning", autoplay=True, height=512, width=512, elem_id="gr-video")
+            video = gr.Video(label="AnimateDiff-Lightning", autoplay=True, height=256, width=256, elem_id="gr-video")
 
     prompt.submit(fn=generate_image, inputs=[prompt, select_base, select_motion, select_step], outputs=video)
     submit.click(fn=generate_image, inputs=[prompt, select_base, select_motion, select_step], outputs=video)
@@ -168,11 +174,11 @@ with gr.Blocks(css=css) as demo:
             ["Focus: Eiffel Tower (Animate: Clouds moving)"],
             ["Focus: Trees In forest (Animate: Lion running)"],
             ["Focus: Astronaut in Space"],
-            ["Focus: Group of Birds in sky (Animate:  Birds Moving) (Shot From distance)"],
-            ["Focus:  Statue of liberty (Shot from Drone) (Animate: Drone coming toward statue)"],
+            ["Focus: Group of Birds in sky (Animate: Birds Moving) (Shot From distance)"],
+            ["Focus: Statue of liberty (Shot from Drone)"],
             ["Focus: Panda in Forest (Animate: Drinking Tea)"],
             ["Focus: Kids Playing (Season: Winter)"],
-            ["Focus: Cars in Street (Season: Rain, Daytime) (Shot from Distance) (Movement: Cars running)"]
+            ["Focus: Cars in Street (Season: Rain, Daytime) (Shot from Distance)"]
         ],
         fn=generate_image,
         inputs=[prompt],