niulx commited on
Commit
fd0bd35
·
verified ·
1 Parent(s): caf4aae

Update segment.py

Browse files
Files changed (1) hide show
  1. segment.py +1 -1
segment.py CHANGED
@@ -13,7 +13,7 @@ import argparse
13
  import matplotlib
14
  import gradio as gr
15
 
16
- @spaces.GPU
17
  def load_image(image_path, left=0, right=0, top=0, bottom=0, size = 512):
18
  if type(image_path) is str:
19
  image = np.array(Image.open(image_path))[:, :, :3]
 
13
  import matplotlib
14
  import gradio as gr
15
 
16
+ @spaces.GPU(duration=10)
17
  def load_image(image_path, left=0, right=0, top=0, bottom=0, size = 512):
18
  if type(image_path) is str:
19
  image = np.array(Image.open(image_path))[:, :, :3]