Update app.py

app.py

@@ -1,5 +1,17 @@
 from spaces import GPU
 import torch
+
+def dummy_warmup():
+    if torch.cuda.is_available():
+        print("[INFO] CUDA is available. Running warmup.")
+        # Run any GPU warm-up or dummy CUDA calls here
+        x = torch.tensor([1.0]).cuda()
+    else:
+        print("[WARNING] CUDA not available. Skipping warmup.")
+
+if __name__ == "__main__":
+    dummy_warmup()
+    
 # import numpy as np
 # import gradio as gr
 # import rembg