bp

app.py

@@ -11,9 +11,15 @@ from huggingface_hub import hf_hub_download
 from PIL import Image
 import matplotlib.pyplot as plt
 from eval_wrapper.eval import EvalWrapper, eval_scene
-from torchvision import transforms
 from spaces import GPU
 
+@GPU
+def dummy_warmup():
+    import torch
+    if torch.cuda.is_available():
+        print("Warmup: GPU is available!")
+        _ = torch.tensor([0.0]).to("cuda")
+
 outdir = "/tmp/rayst3r"
 
 # loading all necessary models