test

modules/masking_module.py

@@ -33,7 +33,7 @@ def load_sam_image_model(
     return SAM2ImagePredictor(sam_model=model)
 
 
-@spaces.GPU(duration=10)
+@spaces.GPU(duration=20)
 def run_sam_inference(
     model: Any,
     image: Image,
@@ -62,7 +62,7 @@ def run_sam_inference(
                 'data':detections,
                 'message':'The AI couldn’t detect the object you want to mask.'
             }
-@spaces.GPU(duration=10)
+@spaces.GPU(duration=20)
 def florence2(image,task_prompt, text_input=None):
     """
     Calling the Microsoft Florence2 model
@@ -130,7 +130,7 @@ def draw_MASK(image, prediction, fill_mask=False):
 
     return new_image
 
-@spaces.GPU(duration=10)
+@spaces.GPU(duration=20)
 def masking_process(image,obj):
     # task_prompt = '<REGION_TO_SEGMENTATION>'
     # # task_prompt = '<OPEN_VOCABULARY_DETECTION>'