ysharma HF Staff commited on
Commit
ed8a3b7
·
1 Parent(s): 9818fa3
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -47,7 +47,7 @@ def generate_transcripts(in_video): #generate_gifs(in_video, gif_transcript):
47
  return transcription, words, words_timestamp
48
 
49
 
50
- def generate_gifs(in_video, gif_transcript, words, words_timestamp):
51
  print("********* Inside generate_gifs() **********")
52
 
53
  #creating list from input gif transcript
@@ -72,7 +72,7 @@ def generate_gifs(in_video, gif_transcript, words, words_timestamp):
72
  print(f"start_seconds, end_seconds are : ({start_seconds}, {end_seconds})")
73
  #generated .gif image
74
  #gif_out, vid_out = gen_moviepy_gif(in_video, start_seconds, end_seconds)
75
- slomo_vid = gen_moviepy_gif(in_video, start_seconds, end_seconds)
76
 
77
  return slomo_vid
78
 
@@ -159,7 +159,7 @@ def get_gif_timestamps(giflist_indxs, words_timestamp):
159
 
160
 
161
  #extracting the video and building and serving a .gif image
162
- def gen_moviepy_gif(in_video, start_seconds, end_seconds):
163
  print("******** inside moviepy_gif () ***************")
164
  #sample
165
  video_path = "./ShiaLaBeouf.mp4"
@@ -169,7 +169,7 @@ def gen_moviepy_gif(in_video, start_seconds, end_seconds):
169
  final_clip = video.subclip(start_seconds, end_seconds)
170
 
171
  #slowmo
172
- slomo_clip = video.subclip(mp.vfx.speedx, 0.5)
173
  slomo_clip.write_videofile("slomo.mp4")
174
 
175
  #writing to RAM
@@ -260,11 +260,12 @@ with demo:
260
  return video[0]
261
 
262
  examples.click(load_examples, examples, input_video)
263
-
 
264
 
265
 
266
  button_transcript.click(generate_transcripts, input_video, [text_transcript, text_words, text_wordstimestamps ])
267
- button_gifs.click(generate_gifs, [input_video, text_gif_transcript, text_words, text_wordstimestamps], out_slomo_vid )
268
 
269
 
270
  demo.launch(debug=True)
 
47
  return transcription, words, words_timestamp
48
 
49
 
50
+ def generate_gifs(in_video, gif_transcript, words, words_timestamp, vid_speed):
51
  print("********* Inside generate_gifs() **********")
52
 
53
  #creating list from input gif transcript
 
72
  print(f"start_seconds, end_seconds are : ({start_seconds}, {end_seconds})")
73
  #generated .gif image
74
  #gif_out, vid_out = gen_moviepy_gif(in_video, start_seconds, end_seconds)
75
+ slomo_vid = gen_moviepy_gif(in_video, start_seconds, end_seconds, vid_speed)
76
 
77
  return slomo_vid
78
 
 
159
 
160
 
161
  #extracting the video and building and serving a .gif image
162
+ def gen_moviepy_gif(in_video, start_seconds, end_seconds, vid_speed):
163
  print("******** inside moviepy_gif () ***************")
164
  #sample
165
  video_path = "./ShiaLaBeouf.mp4"
 
169
  final_clip = video.subclip(start_seconds, end_seconds)
170
 
171
  #slowmo
172
+ slomo_clip = video.subclip(mp.vfx.speedx, vid_speed)
173
  slomo_clip.write_videofile("slomo.mp4")
174
 
175
  #writing to RAM
 
260
  return video[0]
261
 
262
  examples.click(load_examples, examples, input_video)
263
+
264
+ vid_speed = gr.Slider(0.9,0.1)
265
 
266
 
267
  button_transcript.click(generate_transcripts, input_video, [text_transcript, text_words, text_wordstimestamps ])
268
+ button_gifs.click(generate_gifs, [input_video, text_gif_transcript, text_words, text_wordstimestamps, vid_speed], out_slomo_vid )
269
 
270
 
271
  demo.launch(debug=True)