Hugging Face's logo

Spaces:
jotase
/
florence-sam-masking
Running on Zero

App Files Community
jotase commited on
Commit
48ec822
·
verified ·
1 Parent(s): ae4827e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ FLORENCE_MODEL, FLORENCE_PROCESSOR = load_florence_model(device=DEVICE)
23
  SAM_IMAGE_MODEL = load_sam_image_model(device=DEVICE)
24
 
25
 
26
- @spaces.GPU()
27
  @torch.inference_mode()
28
  @torch.autocast(device_type="cuda", dtype=torch.bfloat16)
29
  def process_image(image_input, text_input) -> Optional[Image.Image]:
 
23
  SAM_IMAGE_MODEL = load_sam_image_model(device=DEVICE)
24
 
25
 
26
+ @spaces.GPU(duration=20)
27
  @torch.inference_mode()
28
  @torch.autocast(device_type="cuda", dtype=torch.bfloat16)
29
  def process_image(image_input, text_input) -> Optional[Image.Image]: