killian31 commited on
Commit
f4b4d99
·
1 Parent(s): 7aa414b

fix: temporal coherence

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -15,8 +15,15 @@ def generate_video(audio_path, language, lag, progress=gr.Progress(track_tqdm=Tr
15
  clips = []
16
  total_segments = len(result["segments"])
17
  running_progress = 0.0
 
18
  for segment in result["segments"]:
19
  running_progress += 0.4 / total_segments
 
 
 
 
 
 
20
  text_clip = (
21
  TextClip(
22
  segment["text"],
@@ -30,6 +37,7 @@ def generate_video(audio_path, language, lag, progress=gr.Progress(track_tqdm=Tr
30
  .set_start(segment["start"])
31
  )
32
  clips.append(text_clip)
 
33
  progress(min(0.3 + running_progress, 0.7), "Generating video frames...")
34
 
35
  if lag > 0:
 
15
  clips = []
16
  total_segments = len(result["segments"])
17
  running_progress = 0.0
18
+ current_time = 0.0
19
  for segment in result["segments"]:
20
  running_progress += 0.4 / total_segments
21
+ if segment["start"] > current_time:
22
+ clips.append(
23
+ ColorClip((1280, 720), color=(0, 0, 0)).set_duration(
24
+ segment["start"] - current_time
25
+ )
26
+ )
27
  text_clip = (
28
  TextClip(
29
  segment["text"],
 
37
  .set_start(segment["start"])
38
  )
39
  clips.append(text_clip)
40
+ current_time = segment["end"]
41
  progress(min(0.3 + running_progress, 0.7), "Generating video frames...")
42
 
43
  if lag > 0: