wwen1997 commited on
Commit
622ea3d
1 Parent(s): ca504cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -530,7 +530,7 @@ class Drag:
530
  def reset_states(first_frame_path, last_frame_path, tracking_points):
531
  first_frame_path = gr.State()
532
  last_frame_path = gr.State()
533
- tracking_points = gr.State([])
534
 
535
  return first_frame_path, last_frame_path, tracking_points
536
 
@@ -549,7 +549,7 @@ def preprocess_image(image):
549
 
550
  image_pil.save(first_frame_path)
551
 
552
- return first_frame_path, first_frame_path, gr.State([])
553
 
554
 
555
  def preprocess_image_end(image_end):
@@ -566,7 +566,7 @@ def preprocess_image_end(image_end):
566
 
567
  image_end_pil.save(last_frame_path)
568
 
569
- return last_frame_path, last_frame_path, gr.State([])
570
 
571
 
572
  def add_drag(tracking_points):
 
530
  def reset_states(first_frame_path, last_frame_path, tracking_points):
531
  first_frame_path = gr.State()
532
  last_frame_path = gr.State()
533
+ tracking_points = gr.State()
534
 
535
  return first_frame_path, last_frame_path, tracking_points
536
 
 
549
 
550
  image_pil.save(first_frame_path)
551
 
552
+ return first_frame_path, first_frame_path, gr.State()
553
 
554
 
555
  def preprocess_image_end(image_end):
 
566
 
567
  image_end_pil.save(last_frame_path)
568
 
569
+ return last_frame_path, last_frame_path, gr.State()
570
 
571
 
572
  def add_drag(tracking_points):