bp

app.py

@@ -1,3 +1,12 @@
+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")
+dummy_warmup()
+
 import numpy as np
 import gradio as gr
 import torch
@@ -11,14 +20,7 @@ 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 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"