Spaces:
Sleeping
Sleeping
Commit ·
23d2d0c
1
Parent(s): 274424f
add spaceGPU to func run_inference
Browse files
app.py
CHANGED
|
@@ -110,7 +110,7 @@ def safe_image_input(image):
|
|
| 110 |
|
| 111 |
|
| 112 |
# ====== 推理函数 ======
|
| 113 |
-
|
| 114 |
@torch.no_grad()
|
| 115 |
def run_inference(image_ref, image_tgt, do_rm_bkg):
|
| 116 |
image_ref = safe_image_input(image_ref)
|
|
|
|
| 110 |
|
| 111 |
|
| 112 |
# ====== 推理函数 ======
|
| 113 |
+
@spaces.GPU
|
| 114 |
@torch.no_grad()
|
| 115 |
def run_inference(image_ref, image_tgt, do_rm_bkg):
|
| 116 |
image_ref = safe_image_input(image_ref)
|