jbilcke-hf HF Staff commited on
Commit
699182e
·
1 Parent(s): 48d6121
Files changed (1) hide show
  1. vms/ui/project/tabs/caption_tab.py +2 -1
vms/ui/project/tabs/caption_tab.py CHANGED
@@ -265,7 +265,8 @@ class CaptionTab(BaseTab):
265
  self._should_stop_captioning = False
266
 
267
  try:
268
- copy_files_to_training_dir(prompt_prefix)
 
269
  except Exception as e:
270
  traceback.print_exc()
271
  raise gr.Error(f"Error copying assets to training dir: {str(e)}")
 
265
  self._should_stop_captioning = False
266
 
267
  try:
268
+ # Pass the training_videos_path from the app state
269
+ copy_files_to_training_dir(prompt_prefix, training_videos_path=self.app.training_videos_path)
270
  except Exception as e:
271
  traceback.print_exc()
272
  raise gr.Error(f"Error copying assets to training dir: {str(e)}")