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

Update image_captioning.py

Browse files
Files changed (1) hide show
  1. image_captioning.py +2 -2
image_captioning.py CHANGED
@@ -188,7 +188,7 @@ def encode_video_in_chunks(video_path):
188
  fps = vr.get_avg_fps()
189
 
190
  # Create tmp directory if it doesn't exist
191
- tmp_dir = os.path.join('/teamspace/studios/this_studio', 'tmp')
192
  os.makedirs(tmp_dir, exist_ok=True)
193
 
194
  # Split frame indices into chunks
@@ -281,7 +281,7 @@ def generate_thumbnails(video_path, num_chunks):
281
  total_frames = len(vr)
282
 
283
  # Create/clear tmp directory in current working directory
284
- tmp_dir = os.path.join('/teamspace/studios/this_studio', 'tmp')
285
  # Remove existing directory if it exists
286
  if os.path.exists(tmp_dir):
287
  shutil.rmtree(tmp_dir)
 
188
  fps = vr.get_avg_fps()
189
 
190
  # Create tmp directory if it doesn't exist
191
+ tmp_dir = os.path.join('.', 'tmp')
192
  os.makedirs(tmp_dir, exist_ok=True)
193
 
194
  # Split frame indices into chunks
 
281
  total_frames = len(vr)
282
 
283
  # Create/clear tmp directory in current working directory
284
+ tmp_dir = os.path.join('.', 'tmp')
285
  # Remove existing directory if it exists
286
  if os.path.exists(tmp_dir):
287
  shutil.rmtree(tmp_dir)