Spaces:
Running
on
Zero
Running
on
Zero
Update gradio_utils/utils.py
Browse files- gradio_utils/utils.py +2 -1
gradio_utils/utils.py
CHANGED
|
@@ -113,7 +113,8 @@ def estimate(model, data):
|
|
| 113 |
model.cuda()
|
| 114 |
with torch.no_grad():
|
| 115 |
return model(**data)
|
| 116 |
-
|
|
|
|
| 117 |
def process(query_img, state,
|
| 118 |
cfg_path='configs/test/1shot_split1.py',
|
| 119 |
checkpoint_path='ckpt/1shot_split1.pth'):
|
|
|
|
| 113 |
model.cuda()
|
| 114 |
with torch.no_grad():
|
| 115 |
return model(**data)
|
| 116 |
+
|
| 117 |
+
@spaces.GPU(duration=30)
|
| 118 |
def process(query_img, state,
|
| 119 |
cfg_path='configs/test/1shot_split1.py',
|
| 120 |
checkpoint_path='ckpt/1shot_split1.pth'):
|