Iceclear commited on
Commit
79c4b07
·
verified ·
1 Parent(s): d4d088c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -243,9 +243,9 @@ def generation_loop(video_path='./test_videos', seed=666, fps_out=12, batch_size
243
  prompts = {}
244
  video_list = os.listdir(video_path)
245
  for f in video_list:
246
- if f.endswith(".mp4"):
247
- original_videos.append(f)
248
- prompts[f] = positive_text
249
  print(f"Total prompts to be generated: {len(original_videos)}")
250
  return original_videos, prompts, negative_text
251
 
 
243
  prompts = {}
244
  video_list = os.listdir(video_path)
245
  for f in video_list:
246
+ # if f.endswith(".mp4"):
247
+ original_videos.append(f)
248
+ prompts[f] = positive_text
249
  print(f"Total prompts to be generated: {len(original_videos)}")
250
  return original_videos, prompts, negative_text
251