fffiloni commited on
Commit
dfc461b
Β·
1 Parent(s): 3e7eefe

fix bad video file name + only 1 min lyrics time fo faster inference

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -36,7 +36,7 @@ compel = Compel(
36
 
37
  from pydub import AudioSegment
38
 
39
- def cut_audio(input_path, output_path, max_duration=30000):
40
  audio = AudioSegment.from_file(input_path)
41
 
42
  if len(audio) > max_duration:
@@ -73,6 +73,7 @@ def get_visualizer_video(audio_in, image_in, song_title):
73
  title, # str in 'title' Textbox component
74
  audio_in, # str (filepath or URL to file) in 'audio_in' Audio component
75
  image_in, # str (filepath or URL to image) in 'image_in' Image component
 
76
  api_name="/predict"
77
  )
78
 
@@ -81,7 +82,7 @@ def get_visualizer_video(audio_in, image_in, song_title):
81
  def infer(audio_file, has_lyrics):
82
  print("NEW INFERENCE ...")
83
  gr.Info('Truncating your audio to the first 30 seconds')
84
- truncated_audio = cut_audio(audio_file, "trunc_audio.mp3")
85
  processed_audio = truncated_audio
86
 
87
  print("Calling LP Music Caps...")
@@ -95,10 +96,12 @@ def infer(audio_file, has_lyrics):
95
  if has_lyrics == "Yes" :
96
  print("""β€”β€”β€”
97
  Getting Lyrics ...
 
98
  """)
 
99
  gr.Info("Getting Lyrics ...")
100
  lyrics_result = lyrics_client.predict(
101
- audio_file, # str (filepath or URL to file) in 'Song input' Audio component
102
  fn_index=0
103
  )
104
  print(f"LYRICS: {lyrics_result}")
 
36
 
37
  from pydub import AudioSegment
38
 
39
+ def cut_audio(input_path, output_path, max_duration):
40
  audio = AudioSegment.from_file(input_path)
41
 
42
  if len(audio) > max_duration:
 
73
  title, # str in 'title' Textbox component
74
  audio_in, # str (filepath or URL to file) in 'audio_in' Audio component
75
  image_in, # str (filepath or URL to image) in 'image_in' Image component
76
+ "my_music_to_image_awesome_video.mp4",
77
  api_name="/predict"
78
  )
79
 
 
82
  def infer(audio_file, has_lyrics):
83
  print("NEW INFERENCE ...")
84
  gr.Info('Truncating your audio to the first 30 seconds')
85
+ truncated_audio = cut_audio(audio_file, "trunc_audio.mp3", 30000)
86
  processed_audio = truncated_audio
87
 
88
  print("Calling LP Music Caps...")
 
96
  if has_lyrics == "Yes" :
97
  print("""β€”β€”β€”
98
  Getting Lyrics ...
99
+ Note: We only take the first minute of the song
100
  """)
101
+ truncated_lyrics = cut_audio(audio_file, "trunc_lyrics.mp3", 60000)
102
  gr.Info("Getting Lyrics ...")
103
  lyrics_result = lyrics_client.predict(
104
+ truncated_lyrics, # str (filepath or URL to file) in 'Song input' Audio component
105
  fn_index=0
106
  )
107
  print(f"LYRICS: {lyrics_result}")