gradio

gradio_app.py

@@ -36,11 +36,6 @@ def download_models():
     else:
         print("✅ Pretrained models already exist.")
 
-
-@spaces.GPU
-def warmup():
-    import torch
-    return {"cuda_available": torch.cuda.is_available()}
 # -----------------------------
 # Step 1: Get Anchor Video
 # -----------------------------
@@ -167,11 +162,6 @@ demo = gr.Blocks()
 
 with demo:
 
-    with gr.Row():
-        model_download_output = gr.Textbox(label="🔄 Model Download Log", lines=2)
-        model_download_button = gr.Button("📦 Download Pretrained Models")
-        model_download_button.click(fn=download_models, outputs=model_download_output)
-
     gr.Markdown("## 🎬 EPiC: Cinematic Camera Control")
 
     with gr.Tabs():
@@ -267,8 +257,5 @@ with demo:
         outputs=[step2_video, step2_logs]
     )
 if __name__ == "__main__":
-    try:
-        warmup()  # will quietly trigger ZeroGPU once
-    except Exception as e:
-        print(f"Warmup failed: {e}")
+    download_models()
     demo.launch(server_name="0.0.0.0", server_port=7860, ssr_mode=False)