rename

app.py

@@ -17,7 +17,7 @@ pipe = None
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 
-@spaces.GPU
+@spaces.GPU(duration=1000)
 def load_pipeline(
     pretrained_model_path="THUDM/CogVideoX-5b",
     ef_net_path="weights/EF_Net.pth",
@@ -77,7 +77,7 @@ def load_pipeline(
     return "Pipeline loaded successfully!"
 
 
-@spaces.GPU
+@spaces.GPU(duration=1000)
 def generate_inbetweening(
     first_image: Image.Image,
     last_image: Image.Image,
@@ -286,4 +286,13 @@ with gr.Blocks(title="Sci-Fi: Frame Inbetweening") as demo:
         )
 
 if __name__ == "__main__":
+    # Automatically load pipeline on startup
+    print("Loading pipeline automatically on startup...")
+    try:
+        load_pipeline()
+        print("Pipeline loaded successfully!")
+    except Exception as e:
+        print(f"Failed to load pipeline on startup: {e}")
+        print("You can manually load it from the Setup tab.")
+
     demo.launch()