Spaces:
Runtime error
Runtime error
David Krajewski
commited on
Commit
·
529e347
1
Parent(s):
d668d8e
changed gpu time
Browse files
app.py
CHANGED
|
@@ -236,7 +236,7 @@ class Drag:
|
|
| 236 |
self.width = width
|
| 237 |
self.model_length = model_length
|
| 238 |
|
| 239 |
-
@spaces.GPU(enable_queue=True, duration=
|
| 240 |
def get_cmp_flow(self, frames, sparse_optical_flow, mask, brush_mask=None):
|
| 241 |
|
| 242 |
'''
|
|
@@ -507,7 +507,7 @@ class Drag:
|
|
| 507 |
|
| 508 |
print("models initialized!")
|
| 509 |
|
| 510 |
-
@spaces.GPU(enable_queue=True, duration=
|
| 511 |
def run(self, first_frame_path, tracking_points, inference_batch_size, motion_brush_mask, motion_brush_viz, ctrl_scale):
|
| 512 |
if not self.cmp or not self.pipeline:
|
| 513 |
self.run_model_init()
|
|
|
|
| 236 |
self.width = width
|
| 237 |
self.model_length = model_length
|
| 238 |
|
| 239 |
+
@spaces.GPU(enable_queue=True, duration=120)
|
| 240 |
def get_cmp_flow(self, frames, sparse_optical_flow, mask, brush_mask=None):
|
| 241 |
|
| 242 |
'''
|
|
|
|
| 507 |
|
| 508 |
print("models initialized!")
|
| 509 |
|
| 510 |
+
@spaces.GPU(enable_queue=True, duration=120)
|
| 511 |
def run(self, first_frame_path, tracking_points, inference_batch_size, motion_brush_mask, motion_brush_viz, ctrl_scale):
|
| 512 |
if not self.cmp or not self.pipeline:
|
| 513 |
self.run_model_init()
|