Spaces:

VAST-AI
/

MIDI-3D

Running on Zero

ameerazam08 commited on 4 days ago

Commit

8473f85

verified ·

1 Parent(s): 3a618ed

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -159,7 +159,7 @@ def split_rgb_mask(rgb_image, seg_image):
     return instance_rgbs, instance_masks, scene_rgbs
-# @spaces.GPU()
 @torch.no_grad()
 @torch.autocast(device_type=DEVICE, dtype=torch.bfloat16)
 def run_segmentation(image_prompts: Any, polygon_refinement: bool) -> Image.Image:

     return instance_rgbs, instance_masks, scene_rgbs
+@spaces.GPU()
 @torch.no_grad()
 @torch.autocast(device_type=DEVICE, dtype=torch.bfloat16)
 def run_segmentation(image_prompts: Any, polygon_refinement: bool) -> Image.Image: