hysts HF staff commited on
Commit
acd57c1
1 Parent(s): e03d871
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -346,11 +346,8 @@ def validate_and_convert_image(image, target_size=(512, 512)):
346
  return image
347
 
348
 
349
- def reset_states(first_frame_path, last_frame_path, tracking_points):
350
- first_frame_path = None
351
- last_frame_path = None
352
- tracking_points = []
353
- return first_frame_path, last_frame_path, tracking_points
354
 
355
 
356
  def preprocess_image(image):
@@ -809,8 +806,7 @@ if __name__ == "__main__":
809
 
810
  reset_button.click(
811
  fn=reset_states,
812
- inputs=[first_frame_path, last_frame_path, tracking_points],
813
- outputs=[first_frame_path, last_frame_path, tracking_points],
814
  )
815
 
816
  gr.on(
 
346
  return image
347
 
348
 
349
+ def reset_states():
350
+ return None, None, None, None, []
 
 
 
351
 
352
 
353
  def preprocess_image(image):
 
806
 
807
  reset_button.click(
808
  fn=reset_states,
809
+ outputs=[input_image, input_image_end, first_frame_path, last_frame_path, tracking_points],
 
810
  )
811
 
812
  gr.on(