Update app.py
Browse files
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):
|