Update app.py

app.py

@@ -78,40 +78,40 @@ pipeline = TI2VidTwoStagesPipeline(
     local_files_only=False
 )
 
-print("=" * 80)
-print("Warming up pipeline (loading Gemma text encoder)...")
-print("=" * 80)
-
-# Do a dummy warmup to load all models including Gemma
-import tempfile
-import os
-warmup_output = tempfile.mktemp(suffix=".mp4")
-try:
-    pipeline(
-        prompt="warmup",
-        negative_prompt="",
-        output_path=warmup_output,
-        seed=42,
-        height=256,
-        width=256,
-        num_frames=9,
-        frame_rate=8,
-        num_inference_steps=1,
-        cfg_guidance_scale=1.0,
-        images=[],
-        tiling_config=TilingConfig.default(),
-    )
-    # Clean up warmup output
-    if os.path.exists(warmup_output):
-        os.remove(warmup_output)
-except Exception as e:
-    print(f"Warmup completed with note: {e}")
-
-print("=" * 80)
-print("Pipeline fully loaded and ready!")
-print("=" * 80)
-
-
+# print("=" * 80)
+# print("Warming up pipeline (loading Gemma text encoder)...")
+# print("=" * 80)
+
+# # Do a dummy warmup to load all models including Gemma
+# import tempfile
+# import os
+# warmup_output = tempfile.mktemp(suffix=".mp4")
+# try:
+#     pipeline(
+#         prompt="warmup",
+#         negative_prompt="",
+#         output_path=warmup_output,
+#         seed=42,
+#         height=256,
+#         width=256,
+#         num_frames=9,
+#         frame_rate=8,
+#         num_inference_steps=1,
+#         cfg_guidance_scale=1.0,
+#         images=[],
+#         tiling_config=TilingConfig.default(),
+#     )
+#     # Clean up warmup output
+#     if os.path.exists(warmup_output):
+#         os.remove(warmup_output)
+# except Exception as e:
+#     print(f"Warmup completed with note: {e}")
+
+# print("=" * 80)
+# print("Pipeline fully loaded and ready!")
+# print("=" * 80)
+
+@spaces.GPU(duration=300)
 def generate_video(
     input_image,
     prompt: str,
@@ -184,7 +184,6 @@ def generate_video(
 # Create Gradio interface
 with gr.Blocks(title="LTX-2 Image-to-Video") as demo:
     gr.Markdown("# LTX-2 Image-to-Video Generation")
-    gr.Markdown("Transform images into videos using the LTX-2 2-stage pipeline")
 
     with gr.Row():
         with gr.Column():