Zero GPU test

src/utils.py

@@ -4,9 +4,11 @@ import tempfile
 import torch
 from scipy.spatial import Delaunay
 from .build_pipe import *
+import spaces
 
 pipe = build_pipe()
 
+@spaces.GPU
 def generate_terrain(prompt, num_inference_steps, guidance_scale, seed, crop_size, prefix):
     """Generates terrain data (RGB and elevation) from a text prompt."""
     if prefix and not prefix.endswith(' '):
@@ -56,4 +58,4 @@ def generate_and_display(prompt, num_inference_steps, guidance_scale, seed, crop
         mesh.export(temp_file.name)
         file_path = temp_file.name
 
-    return file_path
+    return file_path