Spaces:
Running
on
Zero
Running
on
Zero
Update segment.py
Browse files- 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]
|