Spaces:
Build error
Build error
onescotch
commited on
Commit
•
85d8725
1
Parent(s):
0515b76
extend gpu usage time
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ print(torch.cuda.get_device_name(index))
|
|
27 |
# from main.inference import Inferer
|
28 |
# inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
|
29 |
|
30 |
-
@spaces.GPU(enable_queue=True)
|
31 |
def infer(video_input, in_threshold=0.5, num_people="Single person", render_mesh=False):
|
32 |
from main.inference import Inferer
|
33 |
inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
|
|
|
27 |
# from main.inference import Inferer
|
28 |
# inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
|
29 |
|
30 |
+
@spaces.GPU(enable_queue=True, duration=300)
|
31 |
def infer(video_input, in_threshold=0.5, num_people="Single person", render_mesh=False):
|
32 |
from main.inference import Inferer
|
33 |
inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
|