gpus

gradio_app.py

@@ -38,7 +38,7 @@ def download_models():
 
 
 @spaces.GPU
-def dummy_gpu():
+def warmup():
     import torch
     return {"cuda_available": torch.cuda.is_available()}
 # -----------------------------
@@ -262,7 +262,9 @@ with demo:
         outputs=[step2_video, step2_logs]
     )
 if __name__ == "__main__":
-    print("Current working directory:", os.getcwd())
     download_models()
-    dummy_gpu()  # <--- Add this line
+    try:
+        warmup()  # will quietly trigger ZeroGPU once
+    except Exception as e:
+        print(f"Warmup failed: {e}")
     demo.launch(server_name="0.0.0.0", server_port=7860)