assentian1970 commited on
Commit
3ad738e
·
verified ·
1 Parent(s): a80b741

Update image_captioning.py

Browse files
Files changed (1) hide show
  1. image_captioning.py +1 -1
image_captioning.py CHANGED
@@ -160,7 +160,7 @@ def process_video_chunk(video_frames, model, tokenizer, processor, prompt):
160
  def split_original_video(video_path, chunk_info):
161
  """Split original video into chunks using precise timestamps"""
162
  original_chunks = []
163
- tmp_dir = os.path.join('/teamspace/studios/this_studio', 'tmp')
164
 
165
  for chunk in chunk_info:
166
  output_path = os.path.join(tmp_dir, f"original_chunk_{chunk['chunk_id']}.mp4")
 
160
  def split_original_video(video_path, chunk_info):
161
  """Split original video into chunks using precise timestamps"""
162
  original_chunks = []
163
+ tmp_dir = os.path.join('.', 'tmp')
164
 
165
  for chunk in chunk_info:
166
  output_path = os.path.join(tmp_dir, f"original_chunk_{chunk['chunk_id']}.mp4")