Spaces:
Running
Running
David Krajewski
commited on
Commit
·
d668d8e
1
Parent(s):
762836f
added print
Browse files
app.py
CHANGED
@@ -504,6 +504,9 @@ class Drag:
|
|
504 |
self.pipeline = self.pipeline.to("cuda:0")
|
505 |
self.cmp = self.cmp.to("cuda:0")
|
506 |
|
|
|
|
|
|
|
507 |
@spaces.GPU(enable_queue=True, duration=240)
|
508 |
def run(self, first_frame_path, tracking_points, inference_batch_size, motion_brush_mask, motion_brush_viz, ctrl_scale):
|
509 |
if not self.cmp or not self.pipeline:
|
|
|
504 |
self.pipeline = self.pipeline.to("cuda:0")
|
505 |
self.cmp = self.cmp.to("cuda:0")
|
506 |
|
507 |
+
|
508 |
+
print("models initialized!")
|
509 |
+
|
510 |
@spaces.GPU(enable_queue=True, duration=240)
|
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:
|