David Krajewski commited on
Commit
d668d8e
·
1 Parent(s): 762836f

added print

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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: