spaces.GPU

README.md

@@ -4,7 +4,7 @@ emoji: 🦀
 colorFrom: blue
 colorTo: indigo
 sdk: gradio
-sdk_version: 4.36.1
+sdk_version: 4.38.1
 python_version: '3.12'
 app_file: app.py
 pinned: false

app.py

@@ -174,8 +174,8 @@ def cleanup_tmp(tmp_root: str = "./tmp", expire_seconds: int = 3600) -> None:
         except Exception as e:
             print(f"[cleanup_tmp] failed to remove {path}: {e}")
 
-@torch.no_grad()
 @spaces.GPU
+@torch.no_grad()
 def run_segmentation(
     image_prompts: Any,
     polygon_refinement: bool = True,
@@ -187,8 +187,7 @@ def run_segmentation(
 
         # pre-process the layers and get the xyxy boxes of each layer
         if len(image_prompts["points"]) == 0:
-            gr.Error("No points provided for segmentation. Please add points to the image.")
-            return None
+            raise gr.Error("No points provided for segmentation. Please add points to the image.")
         
         boxes = [
             [
@@ -221,8 +220,8 @@ def run_segmentation(
 
     return seg_map_pil
 
-@torch.no_grad()
 @spaces.GPU
+@torch.no_grad()
 def run_depth_estimation(
     image_prompts: Any,
     seg_image: Union[str, Image.Image],
@@ -333,7 +332,6 @@ def set_random_seed(seed):
     if torch.cuda.is_available():
         torch.cuda.manual_seed_all(seed)
 
-@spaces.GPU
 def export_single_glb_from_outputs(
     outputs,
     fine_scale,
@@ -387,8 +385,8 @@ def get_duration(rgb_image, seg_image, seed, randomize_seed,
     step_duration = 15.0
     return instance_labels.shape[0] * step_duration
 
-@torch.no_grad()
 @spaces.GPU(duration=get_duration)
+@torch.no_grad()
 def run_generation(
     rgb_image: Any,
     seg_image: Union[str, Image.Image],