Update segment.py

segment.py

@@ -13,7 +13,7 @@ import argparse
 import matplotlib
 import gradio as gr
 
-@spaces.GPU(duration=10)
+
 def load_image(image_path, left=0, right=0, top=0, bottom=0, size = 512):
     if type(image_path) is str:
         image = np.array(Image.open(image_path))[:, :, :3]
@@ -99,7 +99,7 @@ def draw_panoptic_segmentation(segmentation, segments_info,save_folder=None, nos
 
 
 
-
+@spaces.GPU(duration=10)
 def run_segmentation(image, name="example_tmp", size = 512, noseg=False):
 
     base_folder_path = "."